torch<3.0.0,>=2.7.1
huggingface_hub<0.34.0,>=0.33.4

[dev]
mypy
black
matplotlib

[test]
mypy
black
