fix format

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

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/optimum/intel/generation/modeling.py b/optimum/intel/generation/modeling.py
@@ -442,4 +442,4 @@ def _export(
     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)
+        return cls._export(*args, **kwargs)
diff --git a/optimum/intel/neural_compressor/modeling_base.py b/optimum/intel/neural_compressor/modeling_base.py
@@ -223,6 +223,7 @@ def generate(self, *args, **kwargs):
     def _export(cls, *args, **kwargs):
         raise NotImplementedError(f"Export not supported for INC model {cls.__name__}")
 
+
 class INCModelForQuestionAnswering(INCModel):
     auto_model_class = AutoModelForQuestionAnswering
     export_feature = "question-answering"