From 711844ecd8b20ae0138a522c5d1c09d14d6118f4 Mon Sep 17 00:00:00 2001 From: lllyasviel Date: Tue, 6 Feb 2024 17:25:28 -0800 Subject: [PATCH] solve #76 --- extensions-builtin/sd_forge_controlnet/scripts/controlnet.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/extensions-builtin/sd_forge_controlnet/scripts/controlnet.py b/extensions-builtin/sd_forge_controlnet/scripts/controlnet.py index 1b9b690f..8f0a7b85 100644 --- a/extensions-builtin/sd_forge_controlnet/scripts/controlnet.py +++ b/extensions-builtin/sd_forge_controlnet/scripts/controlnet.py @@ -202,7 +202,7 @@ class ControlNetForForgeOfficial(scripts.Script): image = HWC3(image) if using_a1111_data: - mask = HWC3(np.asarray(a1111_i2i_mask)) + mask = HWC3(np.asarray(a1111_i2i_mask)) if a1111_i2i_mask is not None else None elif unit.mask_image is not None and (unit.mask_image['image'] > 5).any(): mask = unit.mask_image['image'] elif unit.mask_image is not None and (unit.mask_image['mask'] > 5).any():