Taiyi-LLM / requirements.txt
lingbionlp's picture
Upload 4 files
9022b4b
accelerate==0.21.0
transformers==4.31.0
peft==0.4.0
bitsandbytes==0.39.0
loguru==0.7.0
numpy==1.24.3
pandas==2.0.3
tqdm==4.62.3
deepspeed==0.9.5
tensorboard==2.14.0
sentencepiece==0.1.99
transformers_stream_generator==0.0.4
tiktoken==0.5.1
einops==0.6.1
torch==1.13.0
scipy==1.10.1
gradio==3.44.4
pydantic==1.10.12