torch>=2.1
safetensors
numpy