Transformers documentation

Transformers

You are viewing main version, which requires installation from source. If you'd like regular pip install, checkout the latest stable version (v5.22.2).
Hugging Face's logo
Join the Hugging Face community

and get access to the augmented documentation experience

to get started

Transformers

Transformers 是用于文本、计算机视觉、音频、视频以及多模态模型的最先进机器学习模型的定义框架,同时支持推理和训练。

它集中管理模型定义,使该定义在整个生态系统中达成一致。transformers 是各框架之间的枢纽:只要支持某个模型定义,它就能与大多数训练框架(Axolotl、Unsloth、DeepSpeed、FSDP、PyTorch-Lightning 等)、推理引擎(vLLM、SGLang、TGI 等)以及相邻的建模库(llama.cpp、mlx 等)兼容,这些库都利用了来自 transformers 的模型定义。

我们致力于通过让模型定义保持简单、可定制且高效,来支持最新前沿模型的发展并推动其普及。

在 Hugging Face Hub 上,有超过 100 万个 模型检查点 可供你使用。

今天就到 Hub 上探索,找到合适的模型,并使用 Transformers 立即开始上手。

探索 模型时间线,了解 Transformers 中最新的文本、视觉、音频和多模态模型架构。

功能

Transformers 提供了使用最先进的预训练模型进行推理或训练所需的一切。主要功能包括:

  • Pipeline:一个简单且经过优化的推理类,适用于文本生成、图像分割、自动语音识别、文档问答等多种机器学习任务。
  • Trainer:一个功能全面的训练器,支持混合精度、torch.compile 和 FlashAttention 等特性,用于 PyTorch 模型的训练和分布式训练。
  • generate:使用大型语言模型(LLM)和视觉语言模型(VLM)进行快速文本生成,支持流式输出和多种解码策略。

设计

阅读我们的设计理念,以了解 Transformers 的设计原则。

Transformers 专为开发者、机器学习工程师和研究人员设计。其主要设计原则如下:

  1. 快速且易于使用:每个模型仅由三个主要类(configuration、model 和 preprocessor)实现,并可通过 Pipeline 或 Trainer 快速用于推理或训练。
  2. 预训练模型:使用预训练模型而非从头训练全新模型,可降低碳足迹、计算成本和时间。每个预训练模型都尽可能忠实地复现了原始模型,并提供最先进的性能。
HuggingFace Expert Acceleration Program

学习

如果您是 Transformers 的新手,或者希望进一步了解 transformer 模型,我们建议从 LLM 课程 开始。这门全面的课程涵盖了从 transformer 模型工作原理的基础知识到各种任务中的实际应用。您将学习完整的工作流程,包括构建高质量数据集、微调大型语言模型以及实现推理能力。课程包含理论学习和动手实践练习,帮助您在学习过程中建立对 transformer 模型的扎实基础。

Update on GitHub