From 08fe455aa50aa3066ef6b7d62c27fc2a7668fe03 Mon Sep 17 00:00:00 2001 From: Alexander Date: Fri, 27 Dec 2024 10:29:34 +0400 Subject: [PATCH] Changed the model for im2im --- tools/who_what_benchmark/tests/test_cli_image.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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"), ], )