diff options
| author | AUTOMATIC1111 <16777216c@gmail.com> | 2024-01-20 10:24:59 +0300 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2024-01-20 10:24:59 +0300 |
| commit | 5df56b3980fed9efc4d0ba3a5a52782c34acd9f1 (patch) | |
| tree | 35bada6a0b1e218bdfdaaf4ed3c20c14a6661f93 /modules/processing.py | |
| parent | 9cdd161160a890f7993f27fd3616421eb665877b (diff) | |
| parent | 2cf23099ebb81832a27c8016f14062885f5a9c98 (diff) | |
Merge pull request #14626 from AUTOMATIC1111/hr-button-fix
more Hr button fix
Diffstat (limited to 'modules/processing.py')
| -rw-r--r-- | modules/processing.py | 7 |
1 files changed, 5 insertions, 2 deletions
diff --git a/modules/processing.py b/modules/processing.py index dcc807fe..547ab2f8 100644 --- a/modules/processing.py +++ b/modules/processing.py @@ -1227,8 +1227,11 @@ class StableDiffusionProcessingTxt2Img(StableDiffusionProcessing): 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)
+ if getattr(self, 'txt2img_upscale', False):
+ total_steps = (self.hr_second_pass_steps or self.steps) * state.job_count
+ else:
+ total_steps = (self.steps + (self.hr_second_pass_steps or self.steps)) * state.job_count
+ shared.total_tqdm.updateTotal(total_steps)
state.job_count = state.job_count * 2
state.processing_has_refined_job_count = True
|