From 57d1c325302181bca9992e8c750039339542fead Mon Sep 17 00:00:00 2001 From: William Berman Date: Mon, 13 Feb 2023 09:11:57 -0800 Subject: [PATCH] karlo image variation use kakaobrain upload --- tests/pipelines/unclip/test_unclip_image_variation.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/pipelines/unclip/test_unclip_image_variation.py b/tests/pipelines/unclip/test_unclip_image_variation.py index e320e48d198a..2fd7d091bf1e 100644 --- a/tests/pipelines/unclip/test_unclip_image_variation.py +++ b/tests/pipelines/unclip/test_unclip_image_variation.py @@ -527,7 +527,7 @@ def test_unclip_image_variation_karlo(self): ) pipeline = UnCLIPImageVariationPipeline.from_pretrained( - "fusing/karlo-image-variations-diffusers", torch_dtype=torch.float16 + "kakaobrain/karlo-v1-alpha-image-variations", torch_dtype=torch.float16 ) pipeline = pipeline.to(torch_device) pipeline.set_progress_bar_config(disable=None)