From 03f4c44d95dd417e830dfcc471b3802c43fc1194 Mon Sep 17 00:00:00 2001 From: Dowon Date: Tue, 27 Feb 2024 21:33:15 +0900 Subject: [PATCH] fix: np.zeros_like -> np.full_like --- controlnet_ext/controlnet_ext_forge.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) 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}