torch>=2.1.0
transformers>=4.44.0
safetensors>=0.4.0

[dev]
pytest>=7.4.0
