From 81745c98fbd7e43bac83f825c08a7611983068e1 Mon Sep 17 00:00:00 2001 From: sayakpaul Date: Fri, 19 Jan 2024 08:29:38 +0530 Subject: [PATCH] make config file local too. --- tests/pipelines/stable_diffusion/test_stable_diffusion.py | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/tests/pipelines/stable_diffusion/test_stable_diffusion.py b/tests/pipelines/stable_diffusion/test_stable_diffusion.py index 8854b482dec7..6dd6e2ab9520 100644 --- a/tests/pipelines/stable_diffusion/test_stable_diffusion.py +++ b/tests/pipelines/stable_diffusion/test_stable_diffusion.py @@ -1243,9 +1243,12 @@ def test_download_from_hub(self): assert image_out.shape == (512, 512, 3) def test_download_local(self): - filename = hf_hub_download("runwayml/stable-diffusion-v1-5", filename="v1-5-pruned-emaonly.ckpt") + ckpt_filename = hf_hub_download("runwayml/stable-diffusion-v1-5", filename="v1-5-pruned-emaonly.ckpt") + config_filename = hf_hub_download("runwayml/stable-diffusion-v1-5", filename="v1-inference.yaml") - pipe = StableDiffusionPipeline.from_single_file(filename, torch_dtype=torch.float16) + pipe = StableDiffusionPipeline.from_single_file( + ckpt_filename, config_files={"v1": config_filename}, torch_dtype=torch.float16 + ) pipe.scheduler = DDIMScheduler.from_config(pipe.scheduler.config) pipe.to("cuda")