diff --git a/modules/processing.py b/modules/processing.py index aafb5a3f8..7b83f2fc3 100644 --- a/modules/processing.py +++ b/modules/processing.py @@ -104,6 +104,7 @@ class StableDiffusionProcessing: self.batch_size: int = batch_size self.n_iter: int = n_iter self.steps: int = steps + self.hr_second_pass_steps = 0 self.cfg_scale: float = cfg_scale self.image_cfg_scale = image_cfg_scale self.diffusers_guidance_rescale = diffusers_guidance_rescale @@ -472,6 +473,7 @@ def create_infotext(p: StableDiffusionProcessing, all_prompts, all_seeds, all_su "Image CFG scale": p.image_cfg_scale if p.enable_hr else None, "Denoising strength": p.denoising_strength if p.enable_hr else None, "Refiner start": p.refiner_start if p.enable_hr else None, + "Secondary steps": p.hr_second_pass_steps if p.enable_hr else None, # restore_faces "Face restoration": shared.opts.face_restoration_model if p.restore_faces else None, # sdnext diff --git a/modules/ui.py b/modules/ui.py index dec77a27c..f06aba6a9 100644 --- a/modules/ui.py +++ b/modules/ui.py @@ -490,6 +490,7 @@ def create_ui(startup_timer = None): (hr_scale, "Hires upscale"), (hr_upscaler, "Hires upscaler"), (hr_second_pass_steps, "Hires steps"), + (hr_second_pass_steps, "Secondary steps"), (hr_resize_x, "Hires resize-1"), (hr_resize_y, "Hires resize-2"), *modules.scripts.scripts_txt2img.infotext_fields @@ -839,6 +840,7 @@ def create_ui(startup_timer = None): (hr_scale, "Hires upscale"), (hr_upscaler, "Hires upscaler"), (hr_second_pass_steps, "Hires steps"), + (hr_second_pass_steps, "Secondary steps"), (hr_resize_x, "Hires resize-1"), (hr_resize_y, "Hires resize-2"), (image_cfg_scale, "Image CFG scale"),