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
@@ -71,11 +71,11 @@ class StableDiffusionControlNetInpaintCannyGenerator:
|
|
71 |
|
72 |
normal_image = image_path["image"].convert("RGB").resize((512, 512))
|
73 |
normal_image = np.array(normal_image)
|
74 |
-
normal_image = Image.fromarray(
|
75 |
|
76 |
mask_image = image_path["mask"].convert("RGB").resize((512, 512))
|
77 |
mask_image = np.array(mask_image)
|
78 |
-
mask_image = Image.fromarray(
|
79 |
|
80 |
control_image = self.controlnet_canny_inpaint(image_path=image_path)
|
81 |
pipe = self.load_model(
|
|
|
71 |
|
72 |
normal_image = image_path["image"].convert("RGB").resize((512, 512))
|
73 |
normal_image = np.array(normal_image)
|
74 |
+
normal_image = Image.fromarray(normal_image)
|
75 |
|
76 |
mask_image = image_path["mask"].convert("RGB").resize((512, 512))
|
77 |
mask_image = np.array(mask_image)
|
78 |
+
mask_image = Image.fromarray(mask_image)
|
79 |
|
80 |
control_image = self.controlnet_canny_inpaint(image_path=image_path)
|
81 |
pipe = self.load_model(
|