UI and PNG info improvements

This commit is contained in:
invincibledude 2023-01-22 15:44:59 +03:00
parent b0ae92d605
commit bbb1e35ea2
2 changed files with 5 additions and 0 deletions

View File

@ -789,6 +789,9 @@ class StableDiffusionProcessingTxt2Img(StableDiffusionProcessing):
if self.hr_resize_x == 0 and self.hr_resize_y == 0: 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 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_x = int(self.width * self.hr_scale)
self.hr_upscale_to_y = int(self.height * self.hr_scale) self.hr_upscale_to_y = int(self.height * self.hr_scale)
else: else:

View File

@ -800,6 +800,8 @@ def create_ui():
(hr_resize_x, "Hires resize-1"), (hr_resize_x, "Hires resize-1"),
(hr_resize_y, "Hires resize-2"), (hr_resize_y, "Hires resize-2"),
(hr_sampler_index, "Hires sampling method"), (hr_sampler_index, "Hires sampling method"),
(hr_prompt, "Hires prompt"),
(hr_negative_prompt, "Hires negative prompt"),
*modules.scripts.scripts_txt2img.infotext_fields *modules.scripts.scripts_txt2img.infotext_fields
] ]
parameters_copypaste.add_paste_fields("txt2img", None, txt2img_paste_fields) parameters_copypaste.add_paste_fields("txt2img", None, txt2img_paste_fields)