Repository navigation
Release v0.5.3
Highlights
- Support using
AutoModelForSequenceClassification.from_pretrainedload reward model @xiaoxigua999 a9c482a- The default
value_headname of the reward model has been changed toscore
- The default
- Support RLOO with per-token KL penalty by @zhuzilin in #515
What's Changed
- Fix bug on prm trainer w.r.t no packing samples and ring attn by @zhuzilin in #551
- Fix issue #549 ZeroDivisionError during DPO/SFT/RM/KTO/KD training eval step by @MarxistZ in #561 @xiaoxigua999
New Contributors
Full Changelog: v0.5.2...v0.5.3