torch>=2.1
safetensors>=0.4