Fix serialization tests for torch>=2.6.0

Author: matthewdouglas

tests/helpers.py

@@ -22,7 +22,7 @@ def torch_save_to_buffer(obj):
 
 def torch_load_from_buffer(buffer):
     buffer.seek(0)
-    obj = torch.load(buffer)
+    obj = torch.load(buffer, weights_only=False)
     buffer.seek(0)
     return obj
 

tests/test_linear8bitlt.py

@@ -118,7 +118,7 @@ def test_linear_serialization(
         if not has_fp16_weights:
             assert os.path.getsize(state_path_8bit) < 0.5 * os.path.getsize(state_path)
 
-        new_state_dict = torch.load(state_path_8bit)
+        new_state_dict = torch.load(state_path_8bit, weights_only=False)
 
     new_linear_custom = Linear8bitLt(
         linear.in_features,