extra check for qs_dict['shape'] is None

poedator · poedator · commit 079d7afe3468 · 2023-11-10T21:48:50.000+03:00
diff --git a/bitsandbytes/functional.py b/bitsandbytes/functional.py
@@ -642,7 +642,7 @@ def from_dict(cls, qs_dict: Dict[str, Any], device: torch.device) -> 'QuantState
             blocksize=qs_dict['blocksize'],
             code=qs_dict['quant_map'].to(device),
             dtype=getattr(torch, qs_dict['dtype']),
-            shape=torch.Size(qs_dict['shape']),
+            shape=torch.Size(qs_dict['shape']) if qs_dict['shape'] is not None else None,
             offset=offset,
             state2=state2,
         )

Original file line number	Diff line number	Diff line change
`@@ -642,7 +642,7 @@ def from_dict(cls, qs_dict: Dict[str, Any], device: torch.device) -> 'QuantState`
`642`	`642`	`blocksize=qs_dict['blocksize'],`
`643`	`643`	`code=qs_dict['quant_map'].to(device),`
`644`	`644`	`dtype=getattr(torch, qs_dict['dtype']),`
`645`		`- shape=torch.Size(qs_dict['shape']),`
	`645`	`+ shape=torch.Size(qs_dict['shape']) if qs_dict['shape'] is not None else None,`
`646`	`646`	`offset=offset,`
`647`	`647`	`state2=state2,`
`648`	`648`	`)`