Fix QuantState.as_dict

cyyever · cyyever · commit 58f149ccf6d7 · 2025-09-03T00:29:44.000+08:00
Signed-off-by: cyy &lt;cyyever@outlook.com&gt;
diff --git a/bitsandbytes/functional.py b/bitsandbytes/functional.py
@@ -511,7 +511,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
`@@ -511,7 +511,7 @@ def as_dict(self, packed=False):`
`511`	`511`	`"blocksize": self.blocksize,`
`512`	`512`	`"quant_map": self.code,`
`513`	`513`	`"dtype": str(self.dtype).strip("torch."),`
`514`		`- "shape": tuple(self.shape),`
	`514`	`+ "shape": tuple(self.shape) if self.shape is not None else None,`
`515`	`515`	`}`
`516`	`516`	`if self.nested:`
`517`	`517`	`qs_dict.update(`