Spaces:

hirol
/

controlnetOverMask

Runtime error

hirol commited on May 7, 2023

Commit

eaa0408

•

1 Parent(s): 7bc5c28

Update generate_img.py

Files changed (1) hide show

generate_img.py CHANGED Viewed

@@ -7,7 +7,9 @@ from diffusers.pipelines.stable_diffusion.pipeline_stable_diffusion_controlnet i
 import random
 # model2
-controlnet1 = ControlNetModel.from_pretrained("lllyasviel/control_v11p_sd15_scribble", torch_dtype=torch.float16)
 def make_inpaint_condition(image, image_mask):

 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)
 def make_inpaint_condition(image, image_mask):