github.devotd.
新发现大语言模型基础设施

huggingface/alignment-handbook — 语言模型对齐训练配方合集

huggingface/alignment-handbook 是一个仓库,以脚本和 YAML 配置的形式汇集了 SFT、DPO、ORPO 等语言模型对齐训练流程。

仓库信息huggingface/alignment-handbook ↗

huggingface/alignment-handbook 是一个仓库,以脚本和 YAML 配置的形式汇集了 SFT、DPO、ORPO 等语言模型对齐训练流程。

GitHub 项目说明: Robust recipes to align language models with human and AI preferences

星标
★ 5.7K
复刻
517
语言
Python
许可证
Apache-2.0
查看原始来源 ↗在 GitHub 查看 ↗

huggingface/alignment-handbook 是一个汇集语言模型对齐(alignment)训练配方的仓库,对齐是指让模型的回答朝人类或其他 AI 更偏好的方向变化的训练。目前没有公开的最新版本,仓库使用 Python 编写,并以 Apache-2.0 许可证公开。

scripts 目录中收录了用于模型训练与评估的持续预训练(continued pretraining)、SFT、DPO、ORPO 等流程。持续预训练是在已训练完成的模型上继续学习新知识的阶段,SFT、DPO、ORPO 则是用于对齐模型响应的训练流程。训练脚本、配置和数据处理方法一并提供,因此可以复现微调过程。全模型训练支持 DeepSpeed ZeRO-3 和 LoRA/QLoRA。ZeRO-3 是把训练所需的内存分摊到多台设备上的方式,LoRA/QLoRA 则是在固定大模型主体的情况下只额外训练轻量模块的方式。recipes 目录中包含用于复现 Zephyr 7B 等模型的 YAML 配置,还说明了如何用自有数据集训练聊天模型。该仓库创建于 2023 年 8 月 25 日,目前有 5697 个 star。

不过,仓库没有给出能够证明对齐训练性能的基准数据。新鲜度、技术、质量三项分数均为空,只有动量分数保留下来,且仅为 1.0 这样较低的值。因此,这个仓库可以作为按照对齐训练流程动手尝试的参考资料,但仅凭它很难判断对齐质量。