torch>=2.2.0
transformers>=4.40.0
datasets
numpy
pyyaml

[bench]
vllm>=0.7.0
flash-attn>=2.7.0

[test]
pytest
pytest-cov
