fix: np.zeros_like -> np.full_like

2026-01-26 11:19:53 +00:00 · 2024-02-27 21:33:15 +09:00
parent 9248b79504
commit 03f4c44d95
1 changed files with 1 additions and 2 deletions
--- 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}