aboutsummaryrefslogtreecommitdiffstats
path: root/modules/processing.py
diff options
context:
space:
mode:
authorAUTOMATIC1111 <16777216c@gmail.com>2023-07-08 11:43:06 +0000
committerGitHub <noreply@github.com>2023-07-08 11:43:06 +0000
commit53924aeaf0da56470658ca2a29ce64ff25d71da3 (patch)
tree41d906a5428007ae6743ad98e847aceb25d951c2 /modules/processing.py
parent953147bf6b5f8de6a0effe3e28013ace0b760f75 (diff)
parentb0ec69b360835a901a1aa57df1f7c8c9d55bf31c (diff)
downloadstable-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.py3
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())