fix: weights_only=False for QuantState

test_quantize_shapes.py

@@ -7,7 +7,7 @@ from pathlib import Path
 
 def test_shapes():
     print("Loading quantized test shard...")
-    ckpt = torch.load("/data/models/test_quantize/model.safetensors", map_location="cpu")
+    ckpt = torch.load("/data/models/test_quantize/model.safetensors", map_location="cpu", weights_only=False)
     
     print(f"Loaded {len(ckpt)} tensors\n")