accelerate>=0.19.0
datasets
numpy
rouge
torch>=1.13.0
safetensors
transformers>=4.31.0
peft

[triton]
triton>=2.0.0
