kadirnar commited on
Commit
ef471c6
·
1 Parent(s): 384ecea

Update diffusion_webui/diffusion_models/controlnet/controlnet_inpaint/controlnet_inpaint_canny.py

Browse files
diffusion_webui/diffusion_models/controlnet/controlnet_inpaint/controlnet_inpaint_canny.py CHANGED
@@ -41,6 +41,7 @@ class StableDiffusionControlNetInpaintCannyGenerator:
41
  return self.pipe
42
 
43
  def load_img(self, image_path):
 
44
  image = image_path["image"].convert("RGB").resize((512, 512))
45
  image = np.array(image)
46
  image = Image.fromarray(image)
@@ -78,8 +79,8 @@ class StableDiffusionControlNetInpaintCannyGenerator:
78
  normal_image = image_path["image"].convert("RGB").resize((512, 512))
79
  mask_image = image_path["mask"].convert("RGB").resize((512, 512))
80
 
81
- normal_image = self.load_img(image_path=image_path)
82
- mask_image = self.load_img(image_path=image_path)
83
 
84
  control_image = self.controlnet_canny_inpaint(image_path=image_path)
85
  pipe = self.load_model(
 
41
  return self.pipe
42
 
43
  def load_img(self, image_path):
44
+ breakpoint()
45
  image = image_path["image"].convert("RGB").resize((512, 512))
46
  image = np.array(image)
47
  image = Image.fromarray(image)
 
79
  normal_image = image_path["image"].convert("RGB").resize((512, 512))
80
  mask_image = image_path["mask"].convert("RGB").resize((512, 512))
81
 
82
+ normal_image = self.load_img(image_path=normal_image)
83
+ mask_image = self.load_img(image_path=mask_image)
84
 
85
  control_image = self.controlnet_canny_inpaint(image_path=image_path)
86
  pipe = self.load_model(