diff options
author | AUTOMATIC1111 <16777216c@gmail.com> | 2023-07-08 11:43:06 +0000 |
---|---|---|
committer | GitHub <noreply@github.com> | 2023-07-08 11:43:06 +0000 |
commit | 53924aeaf0da56470658ca2a29ce64ff25d71da3 (patch) | |
tree | 41d906a5428007ae6743ad98e847aceb25d951c2 /modules/processing.py | |
parent | 953147bf6b5f8de6a0effe3e28013ace0b760f75 (diff) | |
parent | b0ec69b360835a901a1aa57df1f7c8c9d55bf31c (diff) | |
download | stable-diffusion-webui-gfx803-53924aeaf0da56470658ca2a29ce64ff25d71da3.tar.gz stable-diffusion-webui-gfx803-53924aeaf0da56470658ca2a29ce64ff25d71da3.tar.bz2 stable-diffusion-webui-gfx803-53924aeaf0da56470658ca2a29ce64ff25d71da3.zip |
Merge pull request #11477 from hako-mikan/master
add `before_hr` script callback
Diffstat (limited to 'modules/processing.py')
-rw-r--r-- | modules/processing.py | 3 |
1 files changed, 3 insertions, 0 deletions
diff --git a/modules/processing.py b/modules/processing.py index 89ad1bd0..21d1492c 100644 --- a/modules/processing.py +++ b/modules/processing.py @@ -1078,6 +1078,9 @@ class StableDiffusionProcessingTxt2Img(StableDiffusionProcessing): sd_models.apply_token_merging(self.sd_model, self.get_token_merging_ratio(for_hr=True))
+ if self.scripts is not None:
+ self.scripts.before_hr(self)
+
samples = self.sampler.sample_img2img(self, samples, noise, self.hr_c, self.hr_uc, steps=self.hr_second_pass_steps or self.steps, image_conditioning=image_conditioning)
sd_models.apply_token_merging(self.sd_model, self.get_token_merging_ratio())
|