From bbb1e35ea29f8cfc6a6db3b28b10bfbc99033c70 Mon Sep 17 00:00:00 2001
From: invincibledude <>
Date: Sun, 22 Jan 2023 15:44:59 +0300
Subject: [PATCH] UI and PNG info improvements

---
 modules/processing.py | 3 +++
 modules/ui.py         | 2 ++
 2 files changed, 5 insertions(+)

diff --git a/modules/processing.py b/modules/processing.py
index b2de8f92b..0cf4771e4 100644
--- a/modules/processing.py
+++ b/modules/processing.py
@@ -789,6 +789,9 @@ class StableDiffusionProcessingTxt2Img(StableDiffusionProcessing):
 
             if self.hr_resize_x == 0 and self.hr_resize_y == 0:
                 self.extra_generation_params["Hires upscale"] = self.hr_scale
+                self.extra_generation_params["Hires sampler"] = self.hr_sampler
+                self.extra_generation_params["Hires prompt"] = self.hr_prompt
+                self.extra_generation_params["Hires negative prompt"] = self.hr_negative_prompt
                 self.hr_upscale_to_x = int(self.width * self.hr_scale)
                 self.hr_upscale_to_y = int(self.height * self.hr_scale)
             else:
diff --git a/modules/ui.py b/modules/ui.py
index 1656e76da..641808899 100644
--- a/modules/ui.py
+++ b/modules/ui.py
@@ -800,6 +800,8 @@ def create_ui():
                 (hr_resize_x, "Hires resize-1"),
                 (hr_resize_y, "Hires resize-2"),
                 (hr_sampler_index, "Hires sampling method"),
+                (hr_prompt, "Hires prompt"),
+                (hr_negative_prompt, "Hires negative prompt"),
                 *modules.scripts.scripts_txt2img.infotext_fields
             ]
             parameters_copypaste.add_paste_fields("txt2img", None, txt2img_paste_fields)