torch>=2.1
numpy>=1.24
safetensors>=0.4