Commit bbb1e35e authored by invincibledude's avatar invincibledude
Browse files

UI and PNG info improvements

parent b0ae92d6

modules/processing.py

+3 −0
Original line number Diff line number Diff line
@@ -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:

modules/ui.py

+2 −0
Original line number Diff line number Diff line
@@ -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)

CRA Git | Maintained and supported by SUSTech CRA and CCSE