From d4b1bd458d7d164d1c9cb5887536bf10475ae3ff Mon Sep 17 00:00:00 2001 From: natanjunges Date: Fri, 30 Dec 2022 02:50:21 -0300 Subject: [PATCH] Support img2img and inpaint --- scripts/main.py | 13 ++++++++++++- 1 file changed, 12 insertions(+), 1 deletion(-) diff --git a/scripts/main.py b/scripts/main.py index ba140ac..cecc899 100644 --- a/scripts/main.py +++ b/scripts/main.py @@ -326,7 +326,18 @@ class Main(scripts.Script): if model != "Random": payload["models"] = [model] - #img2img/inpainting + if hasattr(p, "init_images"): + buffer = io.BytesIO() + p.init_images[0].save(buffer, format="WEBP") + payload["source_image"] = base64.b64encode(buffer.getvalue()).decode() + + if p.image_mask is None: + payload["source_processing"] = "img2img" + else: + payload["source_processing"] = "inpainting" + buffer = io.BytesIO() + p.image_mask.save(buffer, format="WEBP") + payload["source_mask"] = base64.b64encode(buffer.getvalue()).decode() if len(post_processing) > 0: payload["params"]["post_processing"] = post_processing