ColossalAI/applications/Chat/coati/models
yingliu-hpc 1467e3b41b
[coati] add chatglm model (#4539)
* update configuration of chatglm and add support in coati

* add unit test & update chatglm default config & fix bos index issue

* remove chatglm due to oom

* add dataset pkg in requirement-text

* fix parameter issue in test_models

* add ref in tokenize & rm unnessary parts

* separate source & target tokenization in chatglm

* add unit test to chatglm

* fix test dataset issue

* update truncation of chatglm

* fix Colossalai version

* fix colossal ai version in test
2023-08-29 17:58:51 +08:00
..
base [NFC] polish applications/Chat/coati/models/base/actor.py code style (#4248) 2023-07-26 14:12:57 +08:00
bloom [chat] fix bugs and add unit tests (#4213) 2023-08-02 10:17:36 +08:00
chatglm [coati] add chatglm model (#4539) 2023-08-29 17:58:51 +08:00
gpt [chat] fix bugs and add unit tests (#4213) 2023-08-02 10:17:36 +08:00
llama [chat] fix bugs and add unit tests (#4213) 2023-08-02 10:17:36 +08:00
opt [chat] fix bugs and add unit tests (#4213) 2023-08-02 10:17:36 +08:00
__init__.py [chat] fix bugs and add unit tests (#4213) 2023-08-02 10:17:36 +08:00
generation.py [chat] fix bugs and add unit tests (#4213) 2023-08-02 10:17:36 +08:00
lora.py [chat] fix bugs and add unit tests (#4213) 2023-08-02 10:17:36 +08:00
loss.py [chat] fix bugs and add unit tests (#4213) 2023-08-02 10:17:36 +08:00
utils.py [chat] fix bugs and add unit tests (#4213) 2023-08-02 10:17:36 +08:00