mirrors/stable-diffusion-webui
1
Fork 0
mirror of https://github.com/AUTOMATIC1111/stable-diffusion-webui.git synced 2026-01-24 21:32:03 -08:00
Code Issues Projects Releases Packages Wiki Activity

fix incorrect display/calculation for number of steps for hires fix in progress bars

Browse source
This commit is contained in:
AUTOMATIC 2023-01-04 23:56:43 +03:00
parent 8149078094
commit bc43293c64
3 changed files with 12 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

9
modules/processing.py
View file

@ -685,10 +685,13 @@ class StableDiffusionProcessingTxt2Img(StableDiffusionProcessing):
def init(self, all_prompts, all_seeds, all_subseeds):
if self.enable_hr:
if state.job_count == -1:
state.job_count = self.n_iter * 2
else:
if not state.processing_has_refined_job_count:
if state.job_count == -1:
state.job_count = self.n_iter
shared.total_tqdm.updateTotal((self.steps + (self.hr_second_pass_steps or self.steps)) * state.job_count)
state.job_count = state.job_count * 2
state.processing_has_refined_job_count = True
if self.hr_resize_x == 0 and self.hr_resize_y == 0:
self.extra_generation_params["Hires upscale"] = self.hr_scale