From 2c907b24c98e101aa0fbaeb08fa812df3acb7235 Mon Sep 17 00:00:00 2001 From: DenOfEquity <166248528+DenOfEquity@users.noreply.github.com> Date: Sat, 31 Aug 2024 21:52:10 +0100 Subject: [PATCH] fix for Batch Upload (gallery) now needs to check for None --- 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 cd68727c..353601b5 100644 --- a/extensions-builtin/sd_forge_controlnet/scripts/controlnet.py +++ b/extensions-builtin/sd_forge_controlnet/scripts/controlnet.py @@ -161,7 +161,7 @@ class ControlNetForForgeOfficial(scripts.Script): logger.info(f'Try to read image: {img_path}') img = np.ascontiguousarray(cv2.imread(img_path)[:, :, ::-1]).copy() mask = None - if len(unit.batch_mask_gallery) > 0: + if unit.batch_mask_gallery is not None and len(unit.batch_mask_gallery) > 0: if len(unit.batch_mask_gallery) >= len(unit.batch_input_gallery): mask_path = unit.batch_mask_gallery[idx]['name'] else: