torch>=2.0.0
transformers>=4.40.0
safetensors>=0.4.0

[dev]
pytest>=7.0.0
matplotlib>=3.7.0
psutil>=5.9.0
