Fix QuantState.as_dict

cyyever · cyyever · commit bcfd75cdfa8f · 2025-08-18T14:32:51.000+08:00
Signed-off-by: cyy &lt;cyyever@outlook.com&gt;
diff --git a/bitsandbytes/functional.py b/bitsandbytes/functional.py
@@ -512,7 +512,7 @@ def as_dict(self, packed=False):
             "blocksize": self.blocksize,
             "quant_map": self.code,
             "dtype": str(self.dtype).strip("torch."),
-            "shape": tuple(self.shape),
+            "shape": tuple(self.shape) if self.shape is not None else None,
         }
         if self.nested:
             qs_dict.update(

Original file line number	Diff line number	Diff line change
`@@ -512,7 +512,7 @@ def as_dict(self, packed=False):`
`512`	`512`	`"blocksize": self.blocksize,`
`513`	`513`	`"quant_map": self.code,`
`514`	`514`	`"dtype": str(self.dtype).strip("torch."),`
`515`		`- "shape": tuple(self.shape),`
	`515`	`+ "shape": tuple(self.shape) if self.shape is not None else None,`
`516`	`516`	`}`
`517`	`517`	`if self.nested:`
`518`	`518`	`qs_dict.update(`