Add repr for QuantizedTransformersModel

huggingface · Nov 25, 2024 · 66dca87 · 66dca87
1 parent 4508702
commit 66dca87
Showing 1 changed file with 3 additions and 0 deletions.
diff --git 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"