From f97023315bd10699012a070d0b8f62c7c23b71b0 Mon Sep 17 00:00:00 2001 From: eaidova Date: Mon, 25 Nov 2024 19:01:57 +0400 Subject: [PATCH] use small model for testing textual inversion --- tests/openvino/test_diffusion.py | 12 ++++++------ 1 file changed, 6 insertions(+), 6 deletions(-) diff --git a/tests/openvino/test_diffusion.py b/tests/openvino/test_diffusion.py index 5f151cb83..fd868eb63 100644 --- a/tests/openvino/test_diffusion.py +++ b/tests/openvino/test_diffusion.py @@ -386,8 +386,8 @@ def test_textual_inversion(self): # for now we only test for stable-diffusion # this is very slow and costly to run right now - model_id = "runwayml/stable-diffusion-v1-5" - ti_id = "sd-concepts-library/cat-toy" + model_id = "hf-internal-testing/tiny-stable-diffusion-torch" + ti_id = "katuni4ka/textual_inversion_cat" inputs = self.generate_inputs() inputs["prompt"] = "A backpack" @@ -630,8 +630,8 @@ def test_textual_inversion(self): # for now we only test for stable-diffusion # this is very slow and costly to run right now - model_id = "runwayml/stable-diffusion-v1-5" - ti_id = "sd-concepts-library/cat-toy" + model_id = "hf-internal-testing/tiny-stable-diffusion-torch" + ti_id = "katuni4ka/textual_inversion_cat" inputs = self.generate_inputs(model_type="stable-diffusion") inputs["prompt"] = "A backpack" @@ -874,8 +874,8 @@ def test_textual_inversion(self): # for now we only test for stable-diffusion # this is very slow and costly to run right now - model_id = "runwayml/stable-diffusion-v1-5" - ti_id = "sd-concepts-library/cat-toy" + model_id = "hf-internal-testing/tiny-stable-diffusion-torch" + ti_id = "katuni4ka/textual_inversion_cat" inputs = self.generate_inputs() inputs["prompt"] = "A backpack"