
alignment-handbook
一键复现主流 LLM 对齐实验,从 SFT 到 DPO 全流程开箱即用。
项目预览
项目数据
项目介绍
Star 增长趋势
本站每日记录该项目 GitHub 星标变化,观测越久曲线越完整
数据来源:GitHub API 逐日快照 · 收录于 2026-08-09 · 当前星标 ★ 5712
技术标签
使用教程
环境要求
- Python 3.11(README 用 uv 指定 --python 3.11 创建虚拟环境)
- 已安装 uv(需按 uv 官方安装指南安装)
- 带 CUDA 12.6 的 GPU 环境(PyTorch 通过 cu126 源安装)
- git-lfs(用于向 Hugging Face Hub 推送模型)
- 可用的 Hugging Face 账号(需要 huggingface-cli login)
安装与启动步骤
-
1创建并激活虚拟环境
用 uv 创建名为 handbook 的 Python 3.11 虚拟环境,激活后升级 pip。未装 uv 需先按官方指南安装。
uv venv handbook --python 3.11 && source handbook/bin/activate && uv pip install --upgrade pip -
2安装指定版本 PyTorch
从 PyTorch cu126 源安装 torch 2.6.0。版本必须精确以复现结果,换硬件请参考官方安装页。
uv pip install torch==2.6.0 --index-url https://download.pytorch.org/whl/cu126 -
3安装项目依赖
在仓库根目录安装 alignment-handbook 其余依赖,命令末尾的点表示从当前目录安装。
uv pip install . -
4安装 Flash Attention 2
安装 flash-attn 2.7.4.post1,必须保留 --no-build-isolation 参数,否则编译会失败。
uv pip install "flash-attn==2.7.4.post1" --no-build-isolation -
5登录 Hugging Face
执行后按提示粘贴 access token,用于下载数据集与上传模型到 Hugging Face Hub。
huggingface-cli login -
6安装 Git LFS
安装 git-lfs 以便后续把训练好的模型推送到 Hugging Face Hub,需要 sudo 权限。
sudo apt-get install git-lfs
如何确认成功
上述命令均无报错执行完毕后,仓库根目录下的 scripts 和 recipes 目录即可查看,说明环境准备完成。
常见问题
Q:还没有安装 uv 怎么办?
A:按 README 给出的 UV 安装指南安装:https://docs.astral.sh/uv/getting-started/installation/ ,装好后再执行创建虚拟环境的命令。
Q:PyTorch 版本能换成别的吗?
A:README 强调精确版本对可复现性很重要,因此建议保持 torch==2.6.0;由于版本依赖硬件,也可参考 PyTorch 官方安装页面自行选择。
Q:flash-attn 安装失败是什么原因?
A:README 明确要求带 --no-build-isolation 参数安装,漏掉该参数容易在构建阶段报错,请按原命令重试。
Q:为什么要登录 Hugging Face?
A:登录后可拉取 Hub 上的数据集与模型,并在训练完成后用 git-lfs 将模型推送到 Hugging Face Hub。
注意事项
- README 强调依赖版本精确(torch 2.6.0、flash-attn 2.7.4.post1)以保证可复现性,不要随意升级。
- Flash Attention 2 对硬件和 CUDA 环境有要求,安装前确认 GPU 环境可用。
- 后续训练方法与配方请查阅仓库中的 scripts 和 recipes 目录,README 未在此处给出具体训练命令。
- git-lfs 通过 apt 安装需要 sudo 权限,无权限时请联系管理员或改用其他安装方式。
核心亮点
- 模块化设计,SFT/DPO/KTO 等算法切换仅需修改配置,无需重写代码
- 基于 Hugging Face 生态,与 transformers、peft 深度集成,支持 LoRA 高效微调
- 提供多个已复现的基准配方(如 Zephyr、Tulu),可直接对比结果,节省调参时间
不足之处
- 文档对自定义数据格式和高级用法的说明不够详尽,新手需参考示例代码
- 部分算法(如 KTO)的稳定性和超参数敏感性仍需更多社区验证
适用场景
- 研究者快速验证新的偏好对齐算法效果
- 企业将开源基座模型微调为符合业务规范的客服或助手
- 高校实验室复现顶会论文中的对齐实验并作为基线
替代项目
TRL (Transformer Reinforcement Learning)、Axolotl、LLaMA-Factory
下一篇:xtuner
同类项目推荐
OpenThoughts-Agent
开源
一套配方搞定 Agent 训练数据与流程
Data recipes and robust infrastructure for training AI agents
neural-structured-learning
开源
给模型加图关系与对抗约束,少标注也能训得更准
Training neural models with structured signals.
trl
开源
几行代码给大模型做 RLHF 对齐与偏好微调
Train transformer language models with reinforcement learning.
Complete-LLM-Finetuning
开源
跟着 Notebook 跑一遍,就能亲手微调大模型
stable-dreambooth
开源
用最少代码微调 Stable Diffusion,几张图定制专属模型。
Dreambooth implementation based on Stable Diffusion with minimal code.
lpwm
开源
用潜在粒子建模视频世界,预测未来更精准
[ICLR 2026 Oral] Latent Particle World Models official repository
Directional-Preference-Alignment
开源
用方向性偏好让模型对齐更精准,训练更稳定
Directional Preference Alignment
VideoTuna
开源
用你的数据定制专属视频生成模型,让 AI 更懂你的风格。
Let's finetune video generation models!
mantis
开源
轻量级时间序列分类基础模型,小样本也能精准分类
Mantis: Lightweight Foundation Model for Time Series Classification