ColossalAI/applications/Chat/requirements.txt

transformers>=4.20.1
tqdm
datasets
loralib
colossalai==0.3.1
torch<2.0.0, >=1.12.1
langchain
tokenizers
fastapi
sse_starlette
wandb
sentencepiece
gpustat
[Coati] first commit (#3283) 2023-03-28 12:25:36 +00:00			`transformers>=4.20.1`
			`tqdm`
			`datasets`
			`loralib`
[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 09:58:51 +00:00			`colossalai==0.3.1`
[Coati] first commit (#3283) 2023-03-28 12:25:36 +00:00			`torch<2.0.0, >=1.12.1`
			`langchain`
			`tokenizers`
			`fastapi`
			`sse_starlette`
			`wandb`
			`sentencepiece`
			`gpustat`