ColossalAI/applications/Chat/coati
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
..
dataset [coati] add chatglm model (#4539) 2023-08-29 17:58:51 +08:00
experience_buffer
experience_maker
kernels
models [coati] add chatglm model (#4539) 2023-08-29 17:58:51 +08:00
quant
ray
trainer [coati] add chatglm model (#4539) 2023-08-29 17:58:51 +08:00
__init__.py