From a5b6a5a3adcc845237d872750ded34240cc6a810 Mon Sep 17 00:00:00 2001
From: Jibaku789 <151478027+Jibaku789@users.noreply.github.com>
Date: Mon, 1 Jan 2024 14:58:55 -0600
Subject: [PATCH] Add inpaint options to img2img.py

---
 modules/img2img.py | 12 ++++++++++++
 1 file changed, 12 insertions(+)

diff --git a/modules/img2img.py b/modules/img2img.py
index 04de8e62c..9e09c0a00 100644
--- a/modules/img2img.py
+++ b/modules/img2img.py
@@ -225,6 +225,18 @@ def img2img(id_task: str, mode: int, prompt: str, negative_prompt: str, prompt_s
     if mask:
         p.extra_generation_params["Mask blur"] = mask_blur
 
+        if inpainting_mask_invert is not None:
+            p.extra_generation_params["Mask mode"] = inpainting_mask_invert
+
+        if inpainting_fill is not None:
+            p.extra_generation_params["Masked content"] = inpainting_fill
+
+        if inpaint_full_res is not None:
+            p.extra_generation_params["Inpaint area"] = inpaint_full_res
+
+        if inpaint_full_res_padding is not None:
+            p.extra_generation_params["Only masked padding, pixels"] = inpaint_full_res_padding
+
     with closing(p):
         if is_batch:
             assert not shared.cmd_opts.hide_ui_dir_config, "Launched with --hide-ui-dir-config, batch img2img disabled"
-- 
GitLab