ColossalAI/applications/Chat/coati/models
Hongxin Liu 842768a174
[chat] refactor model save/load logic (#3654)
* [chat] strategy refactor unwrap model

* [chat] strategy refactor save model

* [chat] add docstr

* [chat] refactor trainer save model

* [chat] fix strategy typing

* [chat] refactor trainer save model

* [chat] update readme

* [chat] fix unit test
2023-04-27 18:41:49 +08:00
..
base [chat] refactor model save/load logic (#3654) 2023-04-27 18:41:49 +08:00
bloom [chat] remove lm model class (#3653) 2023-04-27 15:37:38 +08:00
deberta
gpt [chat] remove lm model class (#3653) 2023-04-27 15:37:38 +08:00
llama [chat] remove lm model class (#3653) 2023-04-27 15:37:38 +08:00
opt [chat] remove lm model class (#3653) 2023-04-27 15:37:38 +08:00
roberta
__init__.py [chat] remove lm model class (#3653) 2023-04-27 15:37:38 +08:00
generation.py
lora.py [chat] remove lm model class (#3653) 2023-04-27 15:37:38 +08:00
loss.py
utils.py