Spaces:
Runtime error
Runtime error
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(
|