Skip to content

GitLab

Unverified Commit f190714e authored by Will Berman's avatar Will Berman Committed by GitHub
Browse files

karlo image variation use kakaobrain upload (#2338)

parent 6cbd7b8b
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
tests/pipelines/unclip/test_unclip_image_variation.py
View file @ f190714e
......@@ -527,7 +527,7 @@ class UnCLIPImageVariationPipelineIntegrationTests(unittest.TestCase):
)
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)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment