Spaces:
Building
on
A10G
Building
on
A10G
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=
|
82 |
-
mask_image = self.load_img(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(
|