Fix: Deployment device problem

marwan2232004 · Oct 12, 2024 · 6dc6fd8 · 6dc6fd8
1 parent dba6e0b
commit 6dc6fd8
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/Inference.py b/Inference.py
@@ -47,7 +47,7 @@ def predict(audio_file):
         dim_feedforward=2048,
     )
 
-    model.load_state_dict(torch.load(model_path, weights_only=True))
+    model.load_state_dict(torch.load(model_path, weights_only=False))
     model.to(device)
     model.eval()