cmedina/agenx-lora-training
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix: weights_only=False for QuantState

Browse Source
This commit is contained in:
Christian Medina
2026-07-03 02:07:09 -04:00
parent 5b2d629d11
commit 1c576aac5d
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
test_quantize_shapes.py
View File

@@ -7,7 +7,7 @@ from pathlib import Path
def test_shapes(): def test_shapes():
print("Loading quantized test shard...") 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") print(f"Loaded {len(ckpt)} tensors\n")