Update DPTNet_eval/DPTNet_quant_sep.py

DPTNet_eval/DPTNet_quant_sep.py

@@ -49,13 +49,21 @@ def load_dpt_model():
         token=speech_sep_token
     )
 
-    # 👇 原本邏輯完全不變
     conf_filterbank, conf_masknet = get_conf()
     model_class = getattr(asteroid_test, "DPTNet")
     model = model_class(**conf_filterbank, **conf_masknet)
     model = torch.quantization.quantize_dynamic(model, {torch.nn.LSTM, torch.nn.Linear}, dtype=torch.qint8)
 
-    state_dict = torch.load(model_path, map_location="cpu")
+    try:
+        state_dict = torch.load(model_path, map_location="cpu", weights_only=False)
+    except pickle.UnpicklingError as e:
+        raise RuntimeError(
+            "模型載入失敗！請確認：\n"
+            "1. 模型來源是否可信\n"
+            "2. 是否為舊版 PyTorch 儲存的模型\n"
+            "3. 嘗試鎖定 PyTorch 版本為 2.5.x"
+        ) from e
+
     model.load_state_dict(state_dict)
     model.eval()
     return model