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
strategies [chat] remove lm model class (#3653) 2023-04-27 15:37:38 +08:00
__init__.py
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
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