From c3bd113a0b969732a324aff415c49ea9ad434707 Mon Sep 17 00:00:00 2001
From: InvincibleDude <81354513+InvincibleDude@users.noreply.github.com>
Date: Sun, 5 Feb 2023 15:24:41 +0000
Subject: [PATCH] Image info fix

---
 modules/processing.py | 10 +++++++---
 1 file changed, 7 insertions(+), 3 deletions(-)

diff --git a/modules/processing.py b/modules/processing.py
index 61e6dfcd3..43fac21c1 100644
--- a/modules/processing.py
+++ b/modules/processing.py
@@ -808,6 +808,13 @@ class StableDiffusionProcessingTxt2Img(StableDiffusionProcessing):
 
     def init(self, all_prompts, all_seeds, all_subseeds):
         if self.enable_hr:
+            if self.hr_sampler != '---':
+                self.extra_generation_params["Hires sampler"] = self.hr_sampler
+
+            if self.hr_prompt != '':
+                self.extra_generation_params["Hires prompt"] = f'({self.hr_prompt.replace(",", ";")})'
+                self.extra_generation_params["Hires negative prompt"] = f'({self.hr_negative_prompt.replace(",", ";")})'
+
             if opts.use_old_hires_fix_width_height and self.applied_old_hires_behavior_to != (self.width, self.height):
                 self.hr_resize_x = self.width
                 self.hr_resize_y = self.height
@@ -819,9 +826,6 @@ 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"] = f'({self.hr_prompt.replace(",", ";")})'
-                self.extra_generation_params["Hires negative prompt"] = f'({self.hr_negative_prompt.replace(",", ";")})'
                 self.hr_upscale_to_x = int(self.width * self.hr_scale)
                 self.hr_upscale_to_y = int(self.height * self.hr_scale)
             else: