Update tools/who_what_benchmark/tests/test_cli_image.py

openvinotoolkit · Jan 23, 2025 · e8c95b7 · e8c95b7
1 parent b06c567
commit e8c95b7
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tools/who_what_benchmark/tests/test_cli_image.py b/tools/who_what_benchmark/tests/test_cli_image.py
@@ -106,7 +106,7 @@ def test_image_model_genai(model_id, model_type):
     if ("stable-diffusion-3" in model_id) and model_type != "text-to-image":
         pytest.skip(reason="SD3 is supported as text to image only")
 
-    if ("flux" in model_id) and model_type != "image-inpainting":
+    if ("flux" in model_id) and model_type == "image-inpainting":
         pytest.skip(reason="FLUX is not yet supported as image inpainting")
 
     with tempfile.TemporaryDirectory() as temp_dir: