From 13377aa1cc9e7365e08f9b6c95910907512b5ca2 Mon Sep 17 00:00:00 2001
From: Ella Charlaix <ella@huggingface.co>
Date: Tue, 9 Jan 2024 14:57:44 +0100
Subject: [PATCH] fix format

---
 optimum/intel/generation/modeling.py             | 2 +-
 optimum/intel/neural_compressor/modeling_base.py | 1 +
 2 files changed, 2 insertions(+), 1 deletion(-)

diff --git a/optimum/intel/generation/modeling.py b/optimum/intel/generation/modeling.py
index 19d5413a3e..b6452fce2b 100644
--- 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)
\ No newline at end of file
+        return cls._export(*args, **kwargs)
diff --git a/optimum/intel/neural_compressor/modeling_base.py b/optimum/intel/neural_compressor/modeling_base.py
index e71a29d3b2..c56b3d5a69 100644
--- 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"