From 09cbcd38afb22921d0a1d2476d7671e210e88c16 Mon Sep 17 00:00:00 2001 From: lllyasviel Date: Fri, 2 Feb 2024 21:15:26 -0800 Subject: [PATCH] Update preprocessor_inpaint.py --- .../forge_preprocessor_inpaint/scripts/preprocessor_inpaint.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/extensions-builtin/forge_preprocessor_inpaint/scripts/preprocessor_inpaint.py b/extensions-builtin/forge_preprocessor_inpaint/scripts/preprocessor_inpaint.py index 1f527bd7..abc765e9 100644 --- a/extensions-builtin/forge_preprocessor_inpaint/scripts/preprocessor_inpaint.py +++ b/extensions-builtin/forge_preprocessor_inpaint/scripts/preprocessor_inpaint.py @@ -100,6 +100,9 @@ class PreprocessorInpaintLama(PreprocessorInpaintOnly): return def __call__(self, input_image, resolution, slider_1=None, slider_2=None, slider_3=None, input_mask=None, **kwargs): + if input_mask is None: + return input_image + H, W, C = input_image.shape raw_color = input_image.copy() raw_mask = input_mask.copy()