/CausalLM-Lit

基于 Lightning 的训练语言模型的框架,目前限定模型架构为 DecoderOnly,方便自定义数据,并且集成 trl 强化学习框架。

Primary LanguagePython

Watchers

No one’s watching this repository yet.