ColossalAI/applications/Chat/requirements.txt

14 lines
154 B
Plaintext

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