跳到主要内容

全部菜单

Transformers

Hugging Face 的模型定义框架

AI4天前
分类
AI
语言
英语
发布于
4天前
复刻
34,619

huggingface/transformers ↗, Python, Apache-2.0, 最近提交 22小时前

Transformers 是 Hugging Face 用来集中存放文本、视觉、音频、视频和多模态模型定义的 Python 库,推理和训练都覆盖。模型定义只有一份,所以这里支持的模型可以直接用在 Axolotl、Unsloth、DeepSpeed、FSDP、PyTorch Lightning 等训练框架,vLLM、SGLang、TGI 等推理引擎,以及 llama.cpp、mlx 等相邻库中,它们读取的是同一份定义。Hugging Face Hub 上基于它的检查点超过 100 万个。在 Python 3.10 和 PyTorch 2.5 以上用 pip 安装 transformers[torch] 即可,许可证为 Apache 2.0。

喜欢这个产品吗?