Skip to content

Commit 31981ea

Browse files
committed
Fix style
1 parent cd3bbb3 commit 31981ea

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

optimum/exporters/openvino/__main__.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -369,7 +369,9 @@ def bitnet_load_hook(self, state_dict, prefix, *args, **kwargs):
369369
if (prefix + "weight") in state_dict and state_dict[prefix + "weight"].dtype != self.weight.dtype:
370370
self.original_weight = state_dict[prefix + "weight"]
371371
w_shape = self.original_weight.shape
372-
state_dict[prefix + "weight"] = torch.empty((w_shape[0] * 4, w_shape[1]), dtype=self.weight.dtype, device="meta")
372+
state_dict[prefix + "weight"] = torch.empty(
373+
(w_shape[0] * 4, w_shape[1]), dtype=self.weight.dtype, device="meta"
374+
)
373375
return state_dict
374376

375377
AutoBitLinear.load_hook = bitnet_load_hook

0 commit comments

Comments
 (0)