diff options
| author | hako-mikan <122196982+hako-mikan@users.noreply.github.com> | 2023-06-28 09:37:08 +0000 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2023-06-28 09:37:08 +0000 |
| commit | b0ec69b360835a901a1aa57df1f7c8c9d55bf31c (patch) | |
| tree | b74a35db59200b7e1374eb421d1dd85ddc38f1c5 /modules/processing.py | |
| parent | 394ffa7b0a7fff3ec484bcd084e673a8b301ccc8 (diff) | |
| download | stable-diffusion-webui-gfx803-b0ec69b360835a901a1aa57df1f7c8c9d55bf31c.tar.gz stable-diffusion-webui-gfx803-b0ec69b360835a901a1aa57df1f7c8c9d55bf31c.tar.bz2 stable-diffusion-webui-gfx803-b0ec69b360835a901a1aa57df1f7c8c9d55bf31c.zip | |
add 'before_hr callback' 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 8da73884..35463c37 100644 --- a/modules/processing.py +++ b/modules/processing.py @@ -1074,6 +1074,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())
|