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

[test]
pytest
parameterized

[triton]
triton==2.0.0
