torch>=2.1
safetensors>=0.4.3
numpy

[test]
pytest>=7
