torch==2.1.0
transformers==4.34.1
datasets==2.14.7
accelerate>=0.21.0
peft>=0.6.0
trl>=0.7.4
gradio>=3.38.0,<4.0.0
scipy
sentencepiece
protobuf
tiktoken
jieba
rouge-chinese
nltk
uvicorn
pydantic
fastapi
sse-starlette
matplotlib
deepspeed==0.14.0
transformers_stream_generator
einops
decorator