use small model for testing textual inversion

huggingface · Nov 25, 2024 · f970233 · f970233
1 parent 6a3b1ba
commit f970233
Showing 1 changed file with 6 additions and 6 deletions.
diff --git 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 <cat-toy> 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 <cat-toy> 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 <cat-toy> backpack"