From 1c576aac5d3f03d3976b449608c912fffa0fb269 Mon Sep 17 00:00:00 2001
From: Christian Medina <37550954+cmedinasoriano@users.noreply.github.com>
Date: Fri, 3 Jul 2026 02:07:09 -0400
Subject: [PATCH] fix: weights_only=False for QuantState

---
 test_quantize_shapes.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/test_quantize_shapes.py b/test_quantize_shapes.py
index 457eaa8..c6696a4 100644
--- a/test_quantize_shapes.py
+++ b/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")