Skip to content

Commit

Permalink
fix log
Browse files Browse the repository at this point in the history
  • Loading branch information
SunMarc committed Dec 12, 2023
1 parent 66e3489 commit 1720bab
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion optimum/gptq/quantizer.py
Original file line number Diff line number Diff line change
Expand Up @@ -221,7 +221,7 @@ def convert_model(self, model: nn.Module):
layers_to_keep = sum(self.modules_in_block_to_quantize, [])
for name in list(layers_to_be_replaced.keys()):
if not any(name.endswith(layer) for layer in layers_to_keep):
logger.info(f"Quantization disabled for {name} (only modules_in_block_to_quantize={modules_in_block_to_quantize} are quantized)")
logger.info(f"Quantization disabled for {name} (only modules_in_block_to_quantize={self.modules_in_block_to_quantize} are quantized)")
del layers_to_be_replaced[name]
self._replace_by_quant_layers(model, layers_to_be_replaced)
return model
Expand Down

0 comments on commit 1720bab

Please sign in to comment.