Fix style

Author: l-bat

optimum/commands/export/openvino.py

@@ -19,8 +19,8 @@
 from typing import TYPE_CHECKING, Optional
 
 from ...exporters import TasksManager
-from ..base import BaseOptimumCLICommand, CommandInfo
 from ...intel.utils.import_utils import DIFFUSERS_IMPORT_ERROR, is_diffusers_available
+from ..base import BaseOptimumCLICommand, CommandInfo
 
 
 logger = logging.getLogger(__name__)
@@ -212,7 +212,6 @@ def run(self):
         library_name = TasksManager.infer_library_from_model(self.args.model)
 
         if library_name == "diffusers" and ov_config and ov_config.quantization_config.get("dataset"):
-
             if not is_diffusers_available():
                 raise ValueError(DIFFUSERS_IMPORT_ERROR.format("Export of diffusers models"))
 
@@ -222,13 +221,11 @@ def run(self):
             class_name = diffusers_config.get("_class_name", None)
 
             if class_name == "LatentConsistencyModelPipeline":
-
                 from optimum.intel import OVLatentConsistencyModelPipeline
 
                 model_cls = OVLatentConsistencyModelPipeline
 
             elif class_name == "StableDiffusionXLPipeline":
-
                 from optimum.intel import OVStableDiffusionXLPipeline
 
                 model_cls = OVStableDiffusionXLPipeline
@@ -239,7 +236,9 @@ def run(self):
             else:
                 raise NotImplementedError(f"Quantization in hybrid mode isn't supported for class {class_name}.")
 
-            model = model_cls.from_pretrained(self.args.model, export=True, quantization_config=ov_config.quantization_config)
+            model = model_cls.from_pretrained(
+                self.args.model, export=True, quantization_config=ov_config.quantization_config
+            )
             model.save_pretrained(self.args.output)
 
         else:

tests/openvino/test_exporters_cli.py

@@ -26,6 +26,7 @@
 
 from optimum.exporters.openvino.__main__ import main_export
 from optimum.intel import (  # noqa
+    OVLatentConsistencyModelPipeline,
     OVModelForAudioClassification,
     OVModelForCausalLM,
     OVModelForFeatureExtraction,
@@ -37,7 +38,6 @@
     OVModelForTokenClassification,
     OVStableDiffusionPipeline,
     OVStableDiffusionXLPipeline,
-    OVLatentConsistencyModelPipeline,
 )
 from optimum.intel.openvino.utils import _HEAD_TO_AUTOMODELS
 from optimum.intel.utils.import_utils import is_openvino_tokenizers_available