From 480f4fd8f92b6834144a9a36f0262a0262b18f00 Mon Sep 17 00:00:00 2001 From: Vladimir Mandic Date: Thu, 26 Oct 2023 07:19:15 -0400 Subject: [PATCH] add missing function --- modules/processing.py | 8 ++++++++ 1 file changed, 8 insertions(+) diff --git a/modules/processing.py b/modules/processing.py index 616533987..08be9a3ae 100644 --- a/modules/processing.py +++ b/modules/processing.py @@ -72,6 +72,14 @@ def apply_overlay(image, paste_loc, index, overlays): return image +def create_binary_mask(image): + if image.mode == 'RGBA' and image.getextrema()[-1] != (255, 255): + image = image.split()[-1].convert("L").point(lambda x: 255 if x > 128 else 0) + else: + image = image.convert('L') + return image + + def txt2img_image_conditioning(sd_model, x, width, height): if sd_model.model.conditioning_key in {'hybrid', 'concat'}: # Inpainting models # The "masked-image" in this case will just be all zeros since the entire image is masked.