accelerate>=0.19.0
datasets
numpy
rouge
torch>=1.13.0
safetensors
transformers>=4.26.1

[llama]
transformers>=4.28.0

[triton]
triton>=2.0.0
