accelerate>=0.26.0
datasets
sentencepiece
numpy
rouge
gekko
torch>=1.13.0
safetensors
transformers>=4.31.0
peft>=0.5.0
tqdm

[quality]
ruff==0.1.5

[test]
pytest
parameterized

[triton]
triton==2.0.0
