torch>=2.0
huggingface_hub>=0.20
numpy

[dev]
pytest
