diff options
| author | AUTOMATIC <16777216c@gmail.com> | 2023-04-29 13:28:51 +0000 |
|---|---|---|
| committer | AUTOMATIC <16777216c@gmail.com> | 2023-04-29 13:28:51 +0000 |
| commit | 7428fb5176ccfd203bddcfa30d75c8df5a772cb4 (patch) | |
| tree | 03859ccc3a3dde582e02d222814c061dd24ee59a /modules/processing.py | |
| parent | e10ee9627233a0e0d7cf97e8b582ce543b48439e (diff) | |
| download | stable-diffusion-webui-gfx803-7428fb5176ccfd203bddcfa30d75c8df5a772cb4.tar.gz stable-diffusion-webui-gfx803-7428fb5176ccfd203bddcfa30d75c8df5a772cb4.tar.bz2 stable-diffusion-webui-gfx803-7428fb5176ccfd203bddcfa30d75c8df5a772cb4.zip | |
add is_hr_pass field for processing
Diffstat (limited to 'modules/processing.py')
| -rw-r--r-- | modules/processing.py | 5 |
1 files changed, 5 insertions, 0 deletions
diff --git a/modules/processing.py b/modules/processing.py index c50784f4..8d7b2462 100644 --- a/modules/processing.py +++ b/modules/processing.py @@ -164,6 +164,7 @@ class StableDiffusionProcessing: self.all_seeds = None
self.all_subseeds = None
self.iteration = 0
+ self.is_hr_pass = False
@property
@@ -883,6 +884,8 @@ class StableDiffusionProcessingTxt2Img(StableDiffusionProcessing): if not self.enable_hr:
return samples
+ self.is_hr_pass = True
+
target_width = self.hr_upscale_to_x
target_height = self.hr_upscale_to_y
@@ -952,6 +955,8 @@ class StableDiffusionProcessingTxt2Img(StableDiffusionProcessing): samples = self.sampler.sample_img2img(self, samples, noise, conditioning, unconditional_conditioning, steps=self.hr_second_pass_steps or self.steps, image_conditioning=image_conditioning)
+ self.is_hr_pass = False
+
return samples
|