diff --git a/optimum/quanto/models/transformers_models.py b/optimum/quanto/models/transformers_models.py index 39a6dc55..68bc69ef 100644 --- a/optimum/quanto/models/transformers_models.py +++ b/optimum/quanto/models/transformers_models.py @@ -58,6 +58,9 @@ def forward(self, *args, **kwargs): def __call__(self, *args, **kwargs): return self._wrapped.forward(*args, **kwargs) + def __repr__(self): + return self._wrapped.__repr__() + @staticmethod def _qmap_name(): return f"{QuantizedTransformersModel.BASE_NAME}_qmap.json"