跳到正文
原文
QGallouedec· @QGallouedec · X·· 2 天前AI 评分56

Hugging Face TRL v1.15 发布,默认启用融合 LM head 优化

AI 导读

TRL v1.15 发布,官方称这是迄今最大的一次优化,融合 LM head 默认开启。峰值显存最多降低 82%,序列长度提升 7 倍,DPO 从 10k 提升到 59k tokens,GRPO 从 29k 提升到 115k,更多细节见 github.com/huggingface/trl/r…。

正文 · AI 翻译

TRL v1.15 已发布。这是我们有史以来最大的一次优化。

融合 LM head,默认开启:

→ 峰值显存最多减少 82%

→ 序列长度提升至 7 倍

→ DPO 从 10k 提升到 59k tokens,GRPO 从 29k 提升到 115k

还有更多…… github.com/huggingface/trl/r…

来源:QGallouedec · x.com