LLaMA-Factory/requirements.txt
ShaneTian 69cb0038f0 Update transformers to 4.36.2 to resolve bug when saving a checkpoint in the multi-node setting.
Former-commit-id: 390f0caf7ff28b75b51eb0f3bba079288e8b66d6
2023-12-20 22:00:41 +08:00

20 lines
229 B
Plaintext

torch>=1.13.1
transformers>=4.36.2
datasets>=2.14.3
accelerate>=0.21.0
peft>=0.7.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