torch>=2.1.0
transformers>=4.40.0
numpy>=1.24.0

[dev]
pytest>=7.0
pytest-cov
Pillow
accelerate

[triton]
triton>=2.1.0
