support diffusers 3.5

Author: openvino-dev-samples

optimum/intel/openvino/modeling_diffusion.py

@@ -121,7 +121,7 @@
 else:
     SanaSprintPipeline = object
 
-if is_diffusers_version(">", "0.34.0"):
+if is_diffusers_version(">=", "0.35.0"):
     from diffusers import FluxKontextPipeline
 else:
     FluxKontextPipeline = object

tests/openvino/test_exporters_cli.py

@@ -33,6 +33,7 @@
 from optimum.intel import (  # noqa
     OVFluxFillPipeline,
     OVFluxPipeline,
+    OVFluxKontextPipeline,
     OVLatentConsistencyModelPipeline,
     OVLTXPipeline,
     OVModelForAudioClassification,
@@ -110,6 +111,7 @@ class OVCLIExportTestCase(unittest.TestCase):
                 ("text-to-image", "stable-diffusion-3"),
                 ("text-to-image", "flux"),
                 ("inpainting", "flux-fill"),
+                ("image-to-image", "flux-kontext"),
                 ("text-to-image", "sana"),
                 ("text-to-video", "ltx-video"),
             ]
@@ -137,6 +139,7 @@ class OVCLIExportTestCase(unittest.TestCase):
         "stable-diffusion-3": 6 if is_tokenizers_version("<", "0.20") or is_openvino_version(">=", "2024.5") else 2,
         "flux": 4 if is_tokenizers_version("<", "0.20") or is_openvino_version(">=", "2024.5") else 0,
         "flux-fill": 4 if is_tokenizers_version("<", "0.20") or is_openvino_version(">=", "2024.5") else 0,
+        "flux-kontext": 4 if is_tokenizers_version("<", "0.20") or is_openvino_version(">=", "2024.5") else 0,
         "llava": 2 if is_tokenizers_version("<", "0.20") or is_openvino_version(">=", "2024.5") else 0,
         "sana": 2 if is_tokenizers_version("<", "0.20.0") or is_openvino_version(">=", "2024.5") else 0,
         "ltx-video": 2 if is_tokenizers_version("<", "0.20.0") or is_openvino_version(">=", "2024.5") else 0,