"src/git@developer.sourcefind.cn:renzhc/diffusers_dcu.git" did not exist on "ac3fc649066df9d347df519b3d0877d41fb847b1"
Unverified Commit 5308cce9 authored by Sayak Paul's avatar Sayak Paul Committed by GitHub
Browse files

[Tests] Test for passing local config file to `from_single_file()` (#6638)

make config file local too.
parent 318556b2
...@@ -1246,9 +1246,12 @@ class StableDiffusionPipelineCkptTests(unittest.TestCase): ...@@ -1246,9 +1246,12 @@ class StableDiffusionPipelineCkptTests(unittest.TestCase):
assert image_out.shape == (512, 512, 3) assert image_out.shape == (512, 512, 3)
def test_download_local(self): 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.scheduler = DDIMScheduler.from_config(pipe.scheduler.config)
pipe.to("cuda") pipe.to("cuda")
......
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