fix(transformers): add missing call implementation

huggingface · Sep 30, 2024 · eb281be · eb281be
1 parent 7b73aae
commit eb281be
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
@@ -56,6 +56,9 @@ def __getattr__(self, name: str) -> Any:
     def forward(self, *args, **kwargs):
         return self._wrapped.forward(*args, **kwargs)
 
+    def __call__(self, *args, **kwargs):
+        return self._wrapped.forward(*args, **kwargs)
+
     @staticmethod
     def _qmap_name():
         return f"{QuantizedTransformersModel.BASE_NAME}_qmap.json"