Skip to content

GitLab

Unverified Commit 04681c1d authored by Ivan Gonzalez's avatar Ivan Gonzalez Committed by GitHub
Browse files

typo IBERT in __repr__ quant_mode (#17398) 

fix #17397
parent 13fd6734
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
src/transformers/models/ibert/quant_modules.py
View file @ 04681c1d
...@@ -150,7 +150,7 @@ class QuantAct(nn.Module): ...@@ -150,7 +150,7 @@ class QuantAct(nn.Module):
def __repr__(self): def __repr__(self):
return ( return (
f"{self.__class__.__name__}(activation_bit={self.activation_bit}, " f"{self.__class__.__name__}(activation_bit={self.activation_bit}, "
f"quant_mode: {self.activation_bit}, Act_min: {self.x_min.item():.2f}, " f"quant_mode: {self.quant_mode}, Act_min: {self.x_min.item():.2f}, "
f"Act_max: {self.x_max.item():.2f})" f"Act_max: {self.x_max.item():.2f})"
) )
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment