From 2b264de6448a398dc1b9d0a82223dd161e196a25 Mon Sep 17 00:00:00 2001
From: hlky <106811348+hlky@users.noreply.github.com>
Date: Thu, 25 Aug 2022 11:40:01 +0100
Subject: [PATCH] Update webui.py

---
 webui.py | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/webui.py b/webui.py
index 6068340..24e5a48 100644
--- a/webui.py
+++ b/webui.py
@@ -810,7 +810,11 @@ def img2img(prompt: str, init_info, mask_mode, ddim_steps: int, sampler_name: st
                     prompt_matrix=prompt_matrix,
                     use_GFPGAN=use_GFPGAN,
                     fp=fp,
-                    do_not_save_grid=True
+                    do_not_save_grid=True,
+                    normalize_prompt_weights=normalize_prompt_weights,
+                    init_img=init_img,
+                    init_mask=init_mask,
+                    keep_mask=keep_mask
                 )
 
                 if initial_seed is None: