Update generate_img.py

generate_img.py

@@ -6,6 +6,10 @@ from pipeline_stable_diffusion_controlnet_inpaint import *
 from diffusers.pipelines.stable_diffusion.pipeline_stable_diffusion_controlnet import *
 import random
 
+# model2
+controlnet1 = ControlNetModel.from_pretrained("lllyasviel/control_v11p_sd15_scribble", torch_dtype=torch.float16)
+
+
 def make_inpaint_condition(image, image_mask):
     image = np.array(image.convert("RGB")).astype(np.float32) / 255.0
     image_mask = np.array(image_mask.convert("L"))