From bf63506e8978ef19979abca9bdabc6ccf7c2f223 Mon Sep 17 00:00:00 2001 From: Manuel Schmid Date: Sun, 3 Mar 2024 15:02:03 +0100 Subject: [PATCH] fix: merge full array shapes, not only single channel of mask --- modules/async_worker.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/modules/async_worker.py b/modules/async_worker.py index c1c04b0..31b7cb0 100644 --- a/modules/async_worker.py +++ b/modules/async_worker.py @@ -324,7 +324,7 @@ def worker(): if (isinstance(inpaint_mask_image_upload['image'], np.ndarray) and isinstance(inpaint_mask_image_upload['mask'], np.ndarray) and inpaint_mask_image_upload['image'].ndim == 3): - inpaint_mask_image_upload = np.maximum(inpaint_mask_image_upload['image'], inpaint_mask_image_upload['mask'][:, :, 0]) + inpaint_mask_image_upload = np.maximum(inpaint_mask_image_upload['image'], inpaint_mask_image_upload['mask']) if isinstance(inpaint_mask_image_upload, np.ndarray) and inpaint_mask_image_upload.ndim == 3: H, W, C = inpaint_image.shape inpaint_mask_image_upload = resample_image(inpaint_mask_image_upload, width=W, height=H)