LLaMA-Factory/requirements.txt
2025-09-14 02:26:39 +08:00

36 lines
465 B
Plaintext

# core deps
transformers>=4.49.0,<=4.56.1,!=4.52.0
datasets>=2.16.0,<=4.0.0
accelerate>=1.3.0,<=1.10.1
peft>=0.14.0,<=0.17.1
trl>=0.8.6,<=0.9.6
# gui
gradio>=4.38.0,<=5.45.0
matplotlib>=3.7.0
tyro<0.9.0
# ops
einops
numpy<2.0.0
pandas>=2.0.0
scipy
# model and tokenizer
sentencepiece
tiktoken
modelscope>=1.14.0
hf-transfer
safetensors<=0.5.3
# python
fire
omegaconf
packaging
protobuf
pyyaml
pydantic<=2.10.6
# api
uvicorn
fastapi
sse-starlette
# media
av
librosa