fix export

huggingface · Jan 9, 2024 · b15b251 · b15b251
1 parent c64025d
commit b15b251
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 1 deletion.
diff --git a/optimum/intel/generation/modeling.py b/optimum/intel/generation/modeling.py
@@ -387,7 +387,7 @@ def _from_pretrained(
         )
 
     @classmethod
-    def _from_transformers(
+    def _export(
         cls,
         model_id: str,
         config: PretrainedConfig,
@@ -437,3 +437,9 @@ def _from_transformers(
             model_dtype=torch_dtype,
             **kwargs,
         )
+
+    @classmethod
+    def _from_transformers(cls, *args, **kwargs):
+        # TODO : add warning when from_pretrained_method is set to cls._export instead of cls._from_transformers when export=True
+        # logger.warning("The method `_from_transformers` is deprecated, please use `_export` instead")
+        return cls._export(*args, **kwargs)
diff --git a/optimum/intel/neural_compressor/modeling_base.py b/optimum/intel/neural_compressor/modeling_base.py
@@ -219,6 +219,9 @@ def generate(self, *args, **kwargs):
             )
         return self.model.generate(*args, **kwargs)
 
+    @classmethod
+    def _export(cls, *args, **kwargs):
+        raise NotImplementedError(f"Export not supported for INC model {cls.__name__}")
 
 class INCModelForQuestionAnswering(INCModel):
     auto_model_class = AutoModelForQuestionAnswering