transformers-code

GitHub
3.9k 512 中等 1 次阅读 今天图像语言模型
AI 解读 由 AI 自动生成,仅供参考

transformers-code 是一个专为 Hugging Face Transformers 库打造的实战学习资源库,旨在通过“手把手”的代码演示与同步视频课程,帮助用户从零掌握大模型应用开发。它有效解决了初学者面对庞大 Transformer 生态时“理论难懂、代码难跑、环境难配”的痛点,提供了一套从基础组件讲解到复杂场景落地的完整路径。

这套资源特别适合 AI 开发者、算法研究人员以及希望深入理解 NLP 技术的学生使用。其内容规划循序渐进:从 Pipeline、Tokenizer 等基础入门,覆盖命名实体识别、机器阅读理解、对话机器人等丰富实战案例;更进阶地探讨了基于 PEFT 的高效微调(如 LoRA、P-tuning)、利用 bitsandbytes 实现的低精度训练(含 QLoRA),以及基于 accelerate 的分布式训练方案。

transformers-code 的独特亮点在于其极强的时效性与工程落地性,代码严格适配主流版本(如 PyTorch 2.2+、Transformers 4.42+),并紧跟 LLaMA2、ChatGLM2 等热门模型的最新训练技巧。配合 B 站与 YouTube 上的同步视频讲解,它将枯燥的技术文档转化为可运行的代码实例,是用户系统构建大模型开发能力的优质指南。

使用场景

某初创公司的算法工程师小李,需要在两周内为客服部门构建一个能自动识别用户意图并生成摘要的智能对话系统。

没有 transformers-code 时

  • 环境配置耗时极长:面对 PyTorch、bitsandbytes、accelerate 等复杂的版本依赖冲突,小李花费了三天时间反复调试环境,甚至因版本不兼容导致代码无法运行。
  • 实战案例缺失:网上教程多停留在理论讲解或简单的文本分类,缺乏针对“检索式对话”和“文本摘要”等具体业务场景的完整代码参考,导致开发无从下手。
  • 微调门槛过高:想要适配公司私有数据,必须全量微调大模型,但受限于显存资源,尝试多次均因内存溢出(OOM)而失败,且不懂如何实施 LoRA 或 QLoRA 等高效微调策略。
  • 分布式训练迷茫:面对多卡训练需求,不清楚如何配置 DeepSpeed 或 accelerate,只能单卡慢速训练,严重拖慢项目进度。

使用 transformers-code 后

  • 一键复现环境:直接复用仓库中经过验证的 requirements 配置(如 torch==2.2.1+cu118),半小时内即可完成环境搭建,彻底告别依赖报错。
  • 场景代码即拿即用:参考“实战演练篇”中关于检索式对话机器人和基于 T5/GLM 的文本摘要代码,快速修改数据接口,两天内便跑通了核心业务流程。
  • 低成本高效微调:依据“高效微调篇”和“低精度训练篇”的教程,成功应用 QLoRA 技术将 LlaMA2-7B 模型在单张消费级显卡上完成微调,显存占用降低 70%。
  • 平滑升级分布式训练:利用仓库提供的 accelerate 集成方案,轻松实现多卡并行训练,模型迭代速度提升了数倍,确保项目按时交付。

transformers-code 通过提供从环境搭建到前沿微调策略的全链路实战代码,将大模型落地周期从数周缩短至数天,极大降低了企业的研发成本与技术门槛。

运行环境要求

操作系统
  • 未说明
GPU
  • 需要 NVIDIA GPU (由 cu118 推断),支持 CUDA 11.8
  • 显存需求视具体模型而定,课程涵盖 LLaMA2-7B 和 ChatGLM2-6B 的 4bit/8bit 训练,建议显存 8GB 以上以支持低精度微调
内存

未说明

依赖
notes该仓库为 Transformers 实战课程代码,涵盖从基础入门到分布式训练、低精度训练(QLoRA)及高效微调(LoRA 等)。核心依赖 torch 需匹配 CUDA 11.8 版本。课程涉及大模型(如 LLaMA2, ChatGLM)的微调,建议使用支持 CUDA 的 Linux 环境以获得最佳兼容性。
python未说明
torch==2.2.1+cu118
transformers==4.42.4
peft==0.11.1
datasets==2.20.0
accelerate==0.32.1
bitsandbytes==0.43.1
faiss-cpu==1.7.4
tensorboard==2.14.0
transformers-code hero image

快速开始

手把手带你实战Transformers

简介

手把手带你实战Transformers课程的代码仓库

代码适配

  • torch==2.2.1+cu118

  • transformers==4.42.4

  • peft==0.11.1

  • datasets==2.20.0

  • accelerate==0.32.1

  • bitsandbytes==0.43.1

  • faiss-cpu==1.7.4

  • tensorboard==2.14.0

课程规划

  • 基础入门篇:Transformers入门,从环境安装到各个基础组件的介绍,包括Pipeline、Tokenizer、Model、Datasets、Evaluate、Trainer,并通过一个最基本的文本分类实例将各个模块进行串讲

  • 实战演练篇:Transformers实战,通过丰富的实战案例对Transformers在NLP任务中的解决方案进行介绍,包括命名实体识别、机器阅读理解、多项选择、文本相似度、检索式对话机器人、掩码语言模型、因果语言模型、摘要生成、生成式对话机器人

  • 高效微调篇:Transformers模型高效微调,以PEFT库为核心,介绍各种常用的参数高效微调方法的原理与实战,包括BitFit、Prompt-tuning、P-tuning、Prefix-Tuning、Lora和IA3

  • 低精度训练篇:Transformers模型低精度训练,基于bitsandbytes库,进行模型的低精度训练,包括LlaMA2-7B和ChatGLM2-6B两个模型的多个不同精度训练的实战演练,包括半精度训练、8bit训练、4bit训练(QLoRA)

  • 分布式训练篇:Transformers模型分布式训练,基于accelerate库讲解transformers模型的分布式训练解决方案,介绍分布式训练的基本原理以及accelerate库的基本使用方式,包括与Deepspeed框架的集成

  • 对齐训练篇: ...

  • 性能优化篇: ...

  • 系统演示篇: ...

课程地址

课程视频发布在B站与YouTube,代码与视频会逐步进行更新,目前课程主要更新在B站,YouTube后续会持续更新

Transformers 基础入门篇 (已更新完成)

Transformers 实战演练篇 (已更新完成)

  • 08 基于 Transformers的 NLP解决方案

  • 09 实战演练之 命名实体识别

  • 10 实战演练之 机器阅读理解(上,过长截断策略)

  • 10 实战演练之 机器阅读理解(下,滑动窗口策略)

  • 11 实战演练之 多项选择

  • 12 实战演练之 文本相似度(上,基于交互策略)

  • 12 实战演练之 文本相似度(下,基于匹配策略)

  • 13 实战演练之 检索式对话机器人

  • 14 实战演练之 预训练模型

  • 15 实战演练篇之 文本摘要(上,基于T5模型)

  • 15 实战演练篇之 文本摘要(下,基于GLM模型)

  • 16 实战演练篇之 生成式对话机器人(基于Bloom)

Transformers 参数高效微调篇 (已更新完成)

Transformers 低精度训练篇(已更新完成)

  • 24 低精度训练与模型下载

  • 25 半精度模型训练(上,基于LLaMA2的半精度模型训练)

  • 25 半精度模型训练(下,基于ChatGLM3的半精度模型训练)

  • 26 量化与8bit模型训练

  • 27 4bit量化与QLoRA模型训练

Transformers 分布式训练篇(已更新完成)

Transformers 番外技能篇

  • 基于Optuna的Transformers模型自动调参

Star History

Star History Chart

请作者喝杯奶茶

常见问题

相似工具推荐

openclaw

OpenClaw 是一款专为个人打造的本地化 AI 助手,旨在让你在自己的设备上拥有完全可控的智能伙伴。它打破了传统 AI 助手局限于特定网页或应用的束缚,能够直接接入你日常使用的各类通讯渠道,包括微信、WhatsApp、Telegram、Discord、iMessage 等数十种平台。无论你在哪个聊天软件中发送消息,OpenClaw 都能即时响应,甚至支持在 macOS、iOS 和 Android 设备上进行语音交互,并提供实时的画布渲染功能供你操控。 这款工具主要解决了用户对数据隐私、响应速度以及“始终在线”体验的需求。通过将 AI 部署在本地,用户无需依赖云端服务即可享受快速、私密的智能辅助,真正实现了“你的数据,你做主”。其独特的技术亮点在于强大的网关架构,将控制平面与核心助手分离,确保跨平台通信的流畅性与扩展性。 OpenClaw 非常适合希望构建个性化工作流的技术爱好者、开发者,以及注重隐私保护且不愿被单一生态绑定的普通用户。只要具备基础的终端操作能力(支持 macOS、Linux 及 Windows WSL2),即可通过简单的命令行引导完成部署。如果你渴望拥有一个懂你

349.3k|★★★☆☆|1周前
Agent开发框架图像

stable-diffusion-webui

stable-diffusion-webui 是一个基于 Gradio 构建的网页版操作界面,旨在让用户能够轻松地在本地运行和使用强大的 Stable Diffusion 图像生成模型。它解决了原始模型依赖命令行、操作门槛高且功能分散的痛点,将复杂的 AI 绘图流程整合进一个直观易用的图形化平台。 无论是希望快速上手的普通创作者、需要精细控制画面细节的设计师,还是想要深入探索模型潜力的开发者与研究人员,都能从中获益。其核心亮点在于极高的功能丰富度:不仅支持文生图、图生图、局部重绘(Inpainting)和外绘(Outpainting)等基础模式,还独创了注意力机制调整、提示词矩阵、负向提示词以及“高清修复”等高级功能。此外,它内置了 GFPGAN 和 CodeFormer 等人脸修复工具,支持多种神经网络放大算法,并允许用户通过插件系统无限扩展能力。即使是显存有限的设备,stable-diffusion-webui 也提供了相应的优化选项,让高质量的 AI 艺术创作变得触手可及。

162.1k|★★★☆☆|1周前
开发框架图像Agent

everything-claude-code

everything-claude-code 是一套专为 AI 编程助手(如 Claude Code、Codex、Cursor 等)打造的高性能优化系统。它不仅仅是一组配置文件,而是一个经过长期实战打磨的完整框架,旨在解决 AI 代理在实际开发中面临的效率低下、记忆丢失、安全隐患及缺乏持续学习能力等核心痛点。 通过引入技能模块化、直觉增强、记忆持久化机制以及内置的安全扫描功能,everything-claude-code 能显著提升 AI 在复杂任务中的表现,帮助开发者构建更稳定、更智能的生产级 AI 代理。其独特的“研究优先”开发理念和针对 Token 消耗的优化策略,使得模型响应更快、成本更低,同时有效防御潜在的攻击向量。 这套工具特别适合软件开发者、AI 研究人员以及希望深度定制 AI 工作流的技术团队使用。无论您是在构建大型代码库,还是需要 AI 协助进行安全审计与自动化测试,everything-claude-code 都能提供强大的底层支持。作为一个曾荣获 Anthropic 黑客大奖的开源项目,它融合了多语言支持与丰富的实战钩子(hooks),让 AI 真正成长为懂上

159.3k|★★☆☆☆|今天
开发框架Agent语言模型

ComfyUI

ComfyUI 是一款功能强大且高度模块化的视觉 AI 引擎,专为设计和执行复杂的 Stable Diffusion 图像生成流程而打造。它摒弃了传统的代码编写模式,采用直观的节点式流程图界面,让用户通过连接不同的功能模块即可构建个性化的生成管线。 这一设计巧妙解决了高级 AI 绘图工作流配置复杂、灵活性不足的痛点。用户无需具备编程背景,也能自由组合模型、调整参数并实时预览效果,轻松实现从基础文生图到多步骤高清修复等各类复杂任务。ComfyUI 拥有极佳的兼容性,不仅支持 Windows、macOS 和 Linux 全平台,还广泛适配 NVIDIA、AMD、Intel 及苹果 Silicon 等多种硬件架构,并率先支持 SDXL、Flux、SD3 等前沿模型。 无论是希望深入探索算法潜力的研究人员和开发者,还是追求极致创作自由度的设计师与资深 AI 绘画爱好者,ComfyUI 都能提供强大的支持。其独特的模块化架构允许社区不断扩展新功能,使其成为当前最灵活、生态最丰富的开源扩散模型工具之一,帮助用户将创意高效转化为现实。

108.3k|★★☆☆☆|1周前
开发框架图像Agent

gemini-cli

gemini-cli 是一款由谷歌推出的开源 AI 命令行工具,它将强大的 Gemini 大模型能力直接集成到用户的终端环境中。对于习惯在命令行工作的开发者而言,它提供了一条从输入提示词到获取模型响应的最短路径,无需切换窗口即可享受智能辅助。 这款工具主要解决了开发过程中频繁上下文切换的痛点,让用户能在熟悉的终端界面内直接完成代码理解、生成、调试以及自动化运维任务。无论是查询大型代码库、根据草图生成应用,还是执行复杂的 Git 操作,gemini-cli 都能通过自然语言指令高效处理。 它特别适合广大软件工程师、DevOps 人员及技术研究人员使用。其核心亮点包括支持高达 100 万 token 的超长上下文窗口,具备出色的逻辑推理能力;内置 Google 搜索、文件操作及 Shell 命令执行等实用工具;更独特的是,它支持 MCP(模型上下文协议),允许用户灵活扩展自定义集成,连接如图像生成等外部能力。此外,个人谷歌账号即可享受免费的额度支持,且项目基于 Apache 2.0 协议完全开源,是提升终端工作效率的理想助手。

100.8k|★★☆☆☆|1周前
插件Agent图像

LLMs-from-scratch

LLMs-from-scratch 是一个基于 PyTorch 的开源教育项目,旨在引导用户从零开始一步步构建一个类似 ChatGPT 的大型语言模型(LLM)。它不仅是同名技术著作的官方代码库,更提供了一套完整的实践方案,涵盖模型开发、预训练及微调的全过程。 该项目主要解决了大模型领域“黑盒化”的学习痛点。许多开发者虽能调用现成模型,却难以深入理解其内部架构与训练机制。通过亲手编写每一行核心代码,用户能够透彻掌握 Transformer 架构、注意力机制等关键原理,从而真正理解大模型是如何“思考”的。此外,项目还包含了加载大型预训练权重进行微调的代码,帮助用户将理论知识延伸至实际应用。 LLMs-from-scratch 特别适合希望深入底层原理的 AI 开发者、研究人员以及计算机专业的学生。对于不满足于仅使用 API,而是渴望探究模型构建细节的技术人员而言,这是极佳的学习资源。其独特的技术亮点在于“循序渐进”的教学设计:将复杂的系统工程拆解为清晰的步骤,配合详细的图表与示例,让构建一个虽小但功能完备的大模型变得触手可及。无论你是想夯实理论基础,还是为未来研发更大规模的模型做准备

90.1k|★★★☆☆|1周前
语言模型图像Agent