Spaces:

hirol
/

controlnetOverMask

Runtime error

hirol commited on May 7, 2023

Commit

67e6a13

1 Parent(s): 2db2ca7

Update generate_img.py

Files changed (1) hide show

generate_img.py CHANGED Viewed

@@ -6,7 +6,7 @@ from pipeline_stable_diffusion_controlnet_inpaint import *
 from diffusers.pipelines.stable_diffusion.pipeline_stable_diffusion_controlnet import *
 import random
 #model1
-controlnet =  ControlNetModel.from_pretrained("lllyasviel/control_v11p_sd15_openpose", torch_dtype=torch.float16)
 pipe = StableDiffusionControlNetInpaintPipeline.from_pretrained(
     "./models/Any-inpainting", controlnet=controlnet, torch_dtype=torch.float16,cache_dir='./models'
 )

 from diffusers.pipelines.stable_diffusion.pipeline_stable_diffusion_controlnet import *
 import random
 #model1
+controlnet =  ControlNetModel.from_pretrained("lllyasviel/control_v11f1p_sd15_depth", torch_dtype=torch.float16)
 pipe = StableDiffusionControlNetInpaintPipeline.from_pretrained(
     "./models/Any-inpainting", controlnet=controlnet, torch_dtype=torch.float16,cache_dir='./models'
 )