diff --git a/controlnet_ext/controlnet_ext_forge.py b/controlnet_ext/controlnet_ext_forge.py index e1ac6ed..f829182 100644 --- a/controlnet_ext/controlnet_ext_forge.py +++ b/controlnet_ext/controlnet_ext_forge.py @@ -58,8 +58,7 @@ class ControlNetExt: return image = np.asarray(p.init_images[0]) - mask = np.zeros_like(image) - mask[:] = 255 + mask = np.full_like(image, fill_value=255) cnet_image = {"image": image, "mask": mask}