fix weights only error

brain-score · Jan 31, 2025 · d2df842 · d2df842
1 parent f6c1ce8
commit d2df842
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/brainscore_vision/models/mobilenet_v2_1_0_128/model.py b/brainscore_vision/models/mobilenet_v2_1_0_128/model.py
@@ -27,7 +27,7 @@ def get_model(name):
     """
     assert name == 'mobilenet_v2_1_0_128'
     preprocessing = functools.partial(load_preprocess_images, image_size=128, preprocess_type='inception')
-    model = torch.load(model_weight_path.as_posix())
+    model = torch.load(model_weight_path.as_posix(), weights_only=False)
     wrapper = MobilenetPytorchWrapper(identifier=name, model=model, preprocessing=preprocessing)
     wrapper.image_size = 128
     return wrapper