Spaces:

XAI
/

TAB4IDC-InterventionDemo

Running

pooyanrg commited on Mar 15

Commit

b39464b

1 Parent(s): 6505170

fix

Files changed (1) hide show

utils/model.py CHANGED Viewed

@@ -193,7 +193,7 @@ class CLIP4IDC(CLIP4IDCPreTrainedModel):
 def init_model(model_path, device):
-    model_state_dict = torch.load(model_path, weights_only=False, map_location='cpu')
     # Prepare model
     cache_dir = ""

 def init_model(model_path, device):
+    model_state_dict = torch.load(model_path, map_location='cpu')
     # Prepare model
     cache_dir = ""