callback postprocess_image_after_composite

This commit is contained in:
w-e-w
2024-01-16 14:18:20 +09:00
parent cb5b335acd
commit c1e04c63b3
2 changed files with 22 additions and 0 deletions

View File

@@ -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)