format

Author: echarlaix

optimum/intel/openvino/modeling_diffusion.py

@@ -28,8 +28,8 @@
     LMSDiscreteScheduler,
     PNDMScheduler,
     StableDiffusionPipeline,
-    StableDiffusionXLPipeline,
     StableDiffusionXLImg2ImgPipeline,
+    StableDiffusionXLPipeline,
 )
 from diffusers.schedulers.scheduling_utils import SCHEDULER_CONFIG_NAME
 from diffusers.utils import CONFIG_NAME, is_invisible_watermark_available

tests/openvino/test_export.py

@@ -25,6 +25,7 @@
 from optimum.exporters.openvino import export_from_model
 from optimum.exporters.tasks import TasksManager
 from optimum.intel import (
+    OVLatentConsistencyModelPipeline,
     OVModelForAudioClassification,
     OVModelForCausalLM,
     OVModelForFeatureExtraction,
@@ -39,7 +40,6 @@
     OVStableDiffusionPipeline,
     OVStableDiffusionXLImg2ImgPipeline,
     OVStableDiffusionXLPipeline,
-    OVLatentConsistencyModelPipeline,
 )
 from optimum.intel.openvino.modeling_base import OVBaseModel
 from optimum.utils.save_utils import maybe_load_preprocessors