ColossalAI/applications/Chat/coati/trainer
Hongxin Liu 6ef7011462
[chat] remove lm model class (#3653)
* [chat] refactor lora

* [chat] remove lm class

* [chat] refactor save model

* [chat] refactor train sft

* [chat] fix ci

* [chat] fix ci
2023-04-27 15:37:38 +08:00
..
callbacks [chat] polish performance evaluator (#3647) 2023-04-26 17:34:59 +08:00
strategies [chat] remove lm model class (#3653) 2023-04-27 15:37:38 +08:00
__init__.py [Coati] first commit (#3283) 2023-03-28 20:25:36 +08:00
base.py [chat] refactor trainer (#3648) 2023-04-26 18:11:49 +08:00
ppo.py [chat] refactor trainer (#3648) 2023-04-26 18:11:49 +08:00
rm.py reconstruct chat trainer and fix training script (#3588) 2023-04-18 16:44:03 +08:00
sft.py [chat] remove lm model class (#3653) 2023-04-27 15:37:38 +08:00
utils.py [chat] refactor trainer (#3648) 2023-04-26 18:11:49 +08:00