Compare commits

...

1 Commits

Author SHA1 Message Date
Dhruv Nair
2dd34512a6 update 2024-02-19 05:11:53 +00:00

View File

@@ -559,17 +559,16 @@ class ControlNetInpaintPipelineSlowTests(unittest.TestCase):
def test_load_local(self): def test_load_local(self):
controlnet = ControlNetModel.from_pretrained("lllyasviel/control_v11p_sd15_canny") controlnet = ControlNetModel.from_pretrained("lllyasviel/control_v11p_sd15_canny")
pipe_1 = StableDiffusionControlNetInpaintPipeline.from_pretrained( pipe_1 = StableDiffusionControlNetInpaintPipeline.from_pretrained(
"runwayml/stable-diffusion-v1-5", safety_checker=None, controlnet=controlnet "runwayml/stable-diffusion-inpainting", safety_checker=None, controlnet=controlnet
) )
controlnet = ControlNetModel.from_single_file( controlnet = ControlNetModel.from_single_file(
"https://huggingface.co/lllyasviel/ControlNet-v1-1/blob/main/control_v11p_sd15_canny.pth" "https://huggingface.co/lllyasviel/ControlNet-v1-1/blob/main/control_v11p_sd15_canny.pth"
) )
pipe_2 = StableDiffusionControlNetInpaintPipeline.from_single_file( pipe_2 = StableDiffusionControlNetInpaintPipeline.from_single_file(
"https://huggingface.co/runwayml/stable-diffusion-v1-5/blob/main/v1-5-pruned-emaonly.safetensors", "https://huggingface.co/runwayml/stable-diffusion-inpainting/blob/main/sd-v1-5-inpainting.ckpt",
safety_checker=None, safety_checker=None,
controlnet=controlnet, controlnet=controlnet,
scheduler_type="pndm",
) )
control_image = load_image( control_image = load_image(
"https://huggingface.co/datasets/hf-internal-testing/diffusers-images/resolve/main/sd_controlnet/bird_canny.png" "https://huggingface.co/datasets/hf-internal-testing/diffusers-images/resolve/main/sd_controlnet/bird_canny.png"