aboutsummaryrefslogtreecommitdiffstats
path: root/modules/processing.py
diff options
context:
space:
mode:
authorw-e-w <40751091+w-e-w@users.noreply.github.com>2024-01-16 05:18:20 +0000
committerw-e-w <40751091+w-e-w@users.noreply.github.com>2024-01-16 05:18:20 +0000
commitc1e04c63b3d2a5102b4cf6deddea337c8a964c53 (patch)
tree977b719ea36079472b7a32450ac49de35ba3b775 /modules/processing.py
parentcb5b335acddd126d4f6c990982816c06beb0d6ae (diff)
downloadstable-diffusion-webui-gfx803-c1e04c63b3d2a5102b4cf6deddea337c8a964c53.tar.gz
stable-diffusion-webui-gfx803-c1e04c63b3d2a5102b4cf6deddea337c8a964c53.tar.bz2
stable-diffusion-webui-gfx803-c1e04c63b3d2a5102b4cf6deddea337c8a964c53.zip
callback postprocess_image_after_composite
Diffstat (limited to 'modules/processing.py')
-rw-r--r--modules/processing.py5
1 files changed, 5 insertions, 0 deletions
diff --git a/modules/processing.py b/modules/processing.py
index dcc807fe..2c3365a0 100644
--- a/modules/processing.py
+++ b/modules/processing.py
@@ -1029,6 +1029,11 @@ def process_images_inner(p: StableDiffusionProcessing) -> Processed:
image = apply_overlay(image, p.paste_to, overlay_image)
+ if p.scripts is not None:
+ pp = scripts.PostprocessImageArgs(image)
+ p.scripts.postprocess_image_after_composite(p, pp)
+ image = pp.image
+
if save_samples:
images.save_image(image, p.outpath_samples, "", p.seeds[i], p.prompts[i], opts.samples_format, info=infotext(i), p=p)