diff --git a/tools/who_what_benchmark/tests/test_cli_image.py b/tools/who_what_benchmark/tests/test_cli_image.py index b0bb7b38a5..2516a221a0 100644 --- a/tools/who_what_benchmark/tests/test_cli_image.py +++ b/tools/who_what_benchmark/tests/test_cli_image.py @@ -69,7 +69,7 @@ def test_image_model_types(model_id, model_type, backend): @pytest.mark.parametrize( ("model_id", "model_type"), [ - ("dreamlike-art/dreamlike-anime-1.0", "image-to-image"), + ("katuni4ka/lcm-tiny-sd", "image-to-image"), ("echarlaix/tiny-random-stable-diffusion-xl", "text-to-image"), ], )