Spaces:

hirol
/

controlnetOverMask

Runtime error

hirol commited on May 17, 2023

Commit

e48f389

1 Parent(s): a08b48c

Update generate_img.py

Files changed (1) hide show

generate_img.py CHANGED Viewed

@@ -7,8 +7,8 @@ from diffusers.pipelines.stable_diffusion.pipeline_stable_diffusion_controlnet i
 import random
 # model2
-controlnet = ControlNetModel.from_pretrained("./models/control_any5_openpose", torch_dtype=torch.float16)
-pipe = StableDiffusionControlNetInpaintPipeline.from_pretrained("./models/Any-inpainting", controlnet=controlnet, torch_dtype=torch.float16)
 pipe.scheduler = UniPCMultistepScheduler.from_config(pipe.scheduler.config)

 import random
 # model2
+controlnet = ControlNetModel.from_pretrained("hirol/control_any5_openpose", torch_dtype=torch.float16)
+pipe = StableDiffusionControlNetInpaintPipeline.from_pretrained("hirol/Any-inpainting", controlnet=controlnet, torch_dtype=torch.float16)
 pipe.scheduler = UniPCMultistepScheduler.from_config(pipe.scheduler.config)