llama-factory / requirements.txt
officialhimanshu595's picture
Upload folder using huggingface_hub
20076b6 verified
raw
history blame contribute delete
227 Bytes
torch>=1.13.1
transformers>=4.36.2
datasets>=2.14.3
accelerate>=0.21.0
peft>=0.7.0
trl>=0.7.6
gradio>=3.38.0,<4.0.0
scipy
einops
sentencepiece
protobuf
jieba
rouge-chinese
nltk
uvicorn
pydantic
fastapi
sse-starlette
matplotlib