fix: np.zeros_like -> np.full_like

This commit is contained in:
Dowon
2024-02-27 21:33:15 +09:00
parent 9248b79504
commit 03f4c44d95

View File

@@ -58,8 +58,7 @@ class ControlNetExt:
return return
image = np.asarray(p.init_images[0]) image = np.asarray(p.init_images[0])
mask = np.zeros_like(image) mask = np.full_like(image, fill_value=255)
mask[:] = 255
cnet_image = {"image": image, "mask": mask} cnet_image = {"image": image, "mask": mask}