Testing for the non-safe-tensors case, since the default is safe-tens…

…ors already

tests/models/deformable_detr/test_modeling_deformable_detr.py

@@ -574,7 +574,7 @@ def test_save_load_low_cpu_mem_usage_checkpoints(self):
         pass
 
     @unittest.skip("No support for low_cpu_mem_usage=True.")
-    def test_save_load_low_cpu_mem_usage_safetensors(self):
+    def test_save_load_low_cpu_mem_usage_no_safetensors(self):
         pass
 
     def test_two_stage_training(self):

tests/models/deta/test_modeling_deta.py

@@ -529,7 +529,7 @@ def test_save_load_low_cpu_mem_usage_checkpoints(self):
         pass
 
     @unittest.skip("No support for low_cpu_mem_usage=True.")
-    def test_save_load_low_cpu_mem_usage_safetensors(self):
+    def test_save_load_low_cpu_mem_usage_no_safetensors(self):
         pass
 
 

tests/models/encodec/test_modeling_encodec.py

@@ -329,7 +329,7 @@ def test_save_load_low_cpu_mem_usage_checkpoints(self):
         pass
 
     @unittest.skip("No support for low_cpu_mem_usage=True.")
-    def test_save_load_low_cpu_mem_usage_safetensors(self):
+    def test_save_load_low_cpu_mem_usage_no_safetensors(self):
         pass
 
     def test_determinism(self):

tests/models/lxmert/test_modeling_lxmert.py

@@ -776,7 +776,7 @@ def test_save_load_low_cpu_mem_usage_checkpoints(self):
         pass
 
     @unittest.skip("No support for low_cpu_mem_usage=True.")
-    def test_save_load_low_cpu_mem_usage_safetensors(self):
+    def test_save_load_low_cpu_mem_usage_no_safetensors(self):
         pass
 
 

tests/models/marian/test_modeling_marian.py

@@ -381,7 +381,7 @@ def test_save_load_low_cpu_mem_usage_checkpoints(self):
         pass
 
     @unittest.skip("No support for low_cpu_mem_usage=True.")
-    def test_save_load_low_cpu_mem_usage_safetensors(self):
+    def test_save_load_low_cpu_mem_usage_no_safetensors(self):
         pass
 
 

tests/models/musicgen/test_modeling_musicgen.py

@@ -709,7 +709,7 @@ def test_save_load_low_cpu_mem_usage_checkpoints(self):
         pass
 
     @unittest.skip("No support for low_cpu_mem_usage=True.")
-    def test_save_load_low_cpu_mem_usage_safetensors(self):
+    def test_save_load_low_cpu_mem_usage_no_safetensors(self):
         pass
 
     # override since changing `output_hidden_states` / `output_attentions` from the top-level model config won't work

tests/models/sew/test_modeling_sew.py

@@ -365,7 +365,7 @@ def test_save_load_low_cpu_mem_usage_checkpoints(self):
         pass
 
     @unittest.skip("No support for low_cpu_mem_usage=True.")
-    def test_save_load_low_cpu_mem_usage_safetensors(self):
+    def test_save_load_low_cpu_mem_usage_no_safetensors(self):
         pass
 
     def test_retain_grad_hidden_states_attentions(self):

tests/models/sew_d/test_modeling_sew_d.py

@@ -469,7 +469,7 @@ def test_save_load_low_cpu_mem_usage_checkpoints(self):
         pass
 
     @unittest.skip("No support for low_cpu_mem_usage=True.")
-    def test_save_load_low_cpu_mem_usage_safetensors(self):
+    def test_save_load_low_cpu_mem_usage_no_safetensors(self):
         pass
 
     @slow

tests/models/timm_backbone/test_modeling_timm_backbone.py

@@ -178,7 +178,7 @@ def test_save_load_low_cpu_mem_usage_checkpoints(self):
         pass
 
     @unittest.skip("No support for low_cpu_mem_usage=True.")
-    def test_save_load_low_cpu_mem_usage_safetensors(self):
+    def test_save_load_low_cpu_mem_usage_no_safetensors(self):
         pass
 
     @unittest.skip("model weights aren't tied in TimmBackbone.")

tests/test_modeling_common.py

@@ -456,15 +456,14 @@ def test_save_load_low_cpu_mem_usage_checkpoints(self):
 
                 self._check_save_load_low_cpu_mem_usage(model_class, saved_model_path)
 
-    @require_safetensors
     @slow
-    def test_save_load_low_cpu_mem_usage_safetensors(self):
+    def test_save_load_low_cpu_mem_usage_no_safetensors(self):
         with tempfile.TemporaryDirectory() as saved_model_path:
             for model_class in self.all_model_classes:
                 config, inputs_dict = self.model_tester.prepare_config_and_inputs_for_common()
                 model_to_save = model_class(config)
 
-                model_to_save.save_pretrained(saved_model_path, safe_serialization=True)
+                model_to_save.save_pretrained(saved_model_path, safe_serialization=False)
                 self._check_save_load_low_cpu_mem_usage(model_class, saved_model_path)
 
     def _check_save_load_low_cpu_mem_usage(self, model_class, saved_model_path):