progressive-llm/requirements.txt
2025-07-10 18:09:14 +09:00

13 lines
271 B
Text

transformers>=4.40.0
accelerate>=0.27.0
peft>=0.11.0
trl>=0.9.0
datasets>=2.18.0
bitsandbytes>=0.43.0
wandb>=0.16.0
pyyaml>=6.0
jsonlines>=4.0.0
scikit-learn>=1.3.0
# flash-attn>=2.5.0 # Install separately with --no-build-isolation
sentencepiece>=0.2.0
protobuf>=4.25.0