torch.load weights only due to torch version 2.6

Author: yarden-sony

tests/pytorch_tests/exporter_tests/test_exporting_qat_models.py

@@ -51,7 +51,7 @@ def get_tmp_filepath(self):
         return tempfile.mkstemp('.pt')[1]
 
     def load_exported_model(self, filepath):
-        return torch.load(filepath)
+        return torch.load(filepath, weights_only=False)
 
     def infer(self, model, images):
         return model(images)
@@ -82,7 +82,7 @@ def export_qat_model(self):
         self.final_model(images[0])
         _final_model_tmp_filepath = tempfile.mkstemp('.pt')[1]
         torch.save(self.final_model, _final_model_tmp_filepath)
-        self.final_model = torch.load(_final_model_tmp_filepath)
+        self.final_model = torch.load(_final_model_tmp_filepath, weights_only=False)
         self.final_model(images[0])
 
         self.filepath = self.get_tmp_filepath()

tests/pytorch_tests/function_tests/test_fully_quantized_exporter.py

@@ -94,7 +94,7 @@ def test_save_and_load_model(self):
         print(f"Float Pytorch .pth Model: {float_model_file}")
 
         model = copy.deepcopy(self.fully_quantized_mbv2)
-        model.load_state_dict(torch.load(model_file))
+        model.load_state_dict(torch.load(model_file, weights_only=False))
         model.eval()
         model(next(self.representative_data_gen()))