Skip to content

Commit a8704d2

Browse files
committed
style
1 parent bc9cb55 commit a8704d2

File tree

1 file changed

+5
-1
lines changed

1 file changed

+5
-1
lines changed

src/transformers/quantizers/quantizer_hqq.py

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -112,7 +112,11 @@ def check_quantized_param(
112112
module, tensor_name = get_module_from_name(model, param_name)
113113

114114
if self.pre_quantized:
115-
return (isinstance(module, torch.nn.Linear) or isinstance(module, HQQLinear)) and tensor_name != "weight" and tensor_name != "bias"
115+
return (
116+
(isinstance(module, torch.nn.Linear) or isinstance(module, HQQLinear))
117+
and tensor_name != "weight"
118+
and tensor_name != "bias"
119+
)
116120
else:
117121
return isinstance(module, torch.nn.Linear) and tensor_name == "weight"
118122

0 commit comments

Comments
 (0)