Fix QuantState.as_dict

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