requirements-train.txt

查看原始文件 · 全部源码

# Install torch==2.8.0 separately: cu126 for RTX 3090/4090, cu128 for RTX 5090 (tutorial 3.1).
transformers==4.57.3
trl==0.24.0
peft==0.18.0
bitsandbytes==0.48.2
accelerate==1.11.0
datasets==4.3.0
pytest==8.4.2