diff --git a/tools/who_what_benchmark/tests/test_cli_image.py b/tools/who_what_benchmark/tests/test_cli_image.py index 156d87e05c..3edcc70636 100644 --- 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: