aboutsummaryrefslogtreecommitdiff
path: root/modules
diff options
context:
space:
mode:
Diffstat (limited to 'modules')
-rw-r--r--modules/processing.py3
-rw-r--r--modules/ui.py2
2 files changed, 5 insertions, 0 deletions
diff --git a/modules/processing.py b/modules/processing.py
index b2de8f92..0cf4771e 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 1656e76d..64180889 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)