Files
ColossalAI/applications/ColossalChat/requirements.txt
Hongxin Liu d54642a263 [application] add lora sft example (#6192)
* [application] add lora sft example

* update requirements

* update readme

* update comment

* update ci
2025-02-18 13:06:38 +08:00

25 lines
293 B
Plaintext
Executable File

transformers==4.39.3
tqdm
datasets==2.14.7
loralib
colossalai>=0.4.7
torch>=2.1.0
langchain
tokenizers
fastapi
sse_starlette
wandb
sentencepiece
gpustat
packaging
autoflake==2.2.1
black==23.9.1
tensorboard
six==1.16.0
datasets
ninja==1.11.1
sentencepiece==0.1.99
flash-attn
tiktoken
jsonlines