fix cli tests

Author: eaidova

optimum/intel/openvino/utils.py

@@ -115,6 +115,7 @@
     "audio-classification": "OVModelForAudioClassification",
     "stable-diffusion": "OVStableDiffusionPipeline",
     "stable-diffusion-xl": "OVStableDiffusionXLPipeline",
+    "stable-diffusion-3": "OVStableDiffusion3Pipeline",
     "pix2struct": "OVModelForPix2Struct",
     "latent-consistency": "OVLatentConsistencyModelPipeline",
     "open_clip_text": "OVModelOpenCLIPText",

tests/openvino/test_diffusion.py

@@ -316,7 +316,7 @@ def test_height_width_properties(self, model_arch: str):
 
         self.assertFalse(ov_pipeline.is_dynamic)
         expected_batch = batch_size * num_images_per_prompt
-        if ov_pipeline.unet is not None and "timestep_cond" not in {
+        if ov_pipeline.unet is None or "timestep_cond" not in {
             inputs.get_any_name() for inputs in ov_pipeline.unet.model.inputs
         }:
             expected_batch *= 2
@@ -542,7 +542,7 @@ def test_height_width_properties(self, model_arch: str):
 
         self.assertFalse(ov_pipeline.is_dynamic)
         expected_batch = batch_size * num_images_per_prompt
-        if ov_pipeline.unet is not None and "timestep_cond" not in {
+        if ov_pipeline.unet is None or "timestep_cond" not in {
             inputs.get_any_name() for inputs in ov_pipeline.unet.model.inputs
         }:
             expected_batch *= 2
@@ -770,7 +770,7 @@ def test_height_width_properties(self, model_arch: str):
 
         self.assertFalse(ov_pipeline.is_dynamic)
         expected_batch = batch_size * num_images_per_prompt
-        if ov_pipeline.unet is not None and "timestep_cond" not in {
+        if ov_pipeline.unet is None or "timestep_cond" not in {
             inputs.get_any_name() for inputs in ov_pipeline.unet.model.inputs
         }:
             expected_batch *= 2

tests/openvino/test_exporters_cli.py

@@ -40,6 +40,7 @@
     OVModelOpenCLIPText,
     OVModelOpenCLIPVisual,
     OVSentenceTransformer,
+    OVStableDiffusion3Pipeline,
     OVStableDiffusionPipeline,
     OVStableDiffusionXLPipeline,
 )

tests/openvino/utils_tests.py

@@ -118,7 +118,7 @@
     "stable-diffusion-openvino": "hf-internal-testing/tiny-stable-diffusion-openvino",
     "stable-diffusion-xl": "echarlaix/tiny-random-stable-diffusion-xl",
     "stable-diffusion-xl-refiner": "echarlaix/tiny-random-stable-diffusion-xl-refiner",
-    "stable-diffusion-3": "katuni4ka/tiny-random-stable-diffusion-3",
+    "stable-diffusion-3": "yujiepan/stable-diffusion-3-tiny-random",
     "stablelm": "hf-internal-testing/tiny-random-StableLmForCausalLM",
     "starcoder2": "hf-internal-testing/tiny-random-Starcoder2ForCausalLM",
     "latent-consistency": "echarlaix/tiny-random-latent-consistency",