fix load state dict

P3-SAM/model.py

@@ -113,7 +113,7 @@ def load_state_dict(self,
                     ignore_seg_s2_mlp=False, 
                     ignore_iou_mlp=False):
     if ckpt_path is not None:
-        state_dict = torch.load(ckpt_path, map_location="cpu")["state_dict"]
+        state_dict = torch.load(ckpt_path, weights_only=False, map_location="cpu")["state_dict"]
     elif state_dict is None:
         # download from huggingface
         print(f'trying to download model from huggingface...')