aboutsummaryrefslogtreecommitdiffstats
path: root/modules/images.py
diff options
context:
space:
mode:
authorAUTOMATIC1111 <16777216c@gmail.com>2022-10-25 05:40:19 +0000
committerGitHub <noreply@github.com>2022-10-25 05:40:19 +0000
commitf53ca516385cddf07c4429720e6bd373547e9112 (patch)
treeebe736abe4098fe25ba8536533b6a7d583124cef /modules/images.py
parent77a320f406a76425176b8ca4c034c362b6734713 (diff)
parent16416e42b5d06c7c580e12dab427e29750146604 (diff)
downloadstable-diffusion-webui-gfx803-f53ca516385cddf07c4429720e6bd373547e9112.tar.gz
stable-diffusion-webui-gfx803-f53ca516385cddf07c4429720e6bd373547e9112.tar.bz2
stable-diffusion-webui-gfx803-f53ca516385cddf07c4429720e6bd373547e9112.zip
Merge pull request #3549 from tsngo/on-image-saved-callback
add callback after image is saved
Diffstat (limited to 'modules/images.py')
-rw-r--r--modules/images.py3
1 files changed, 2 insertions, 1 deletions
diff --git a/modules/images.py b/modules/images.py
index 9a8fe3ed..0e044af2 100644
--- a/modules/images.py
+++ b/modules/images.py
@@ -16,7 +16,7 @@ from PIL import Image, ImageFont, ImageDraw, PngImagePlugin
from fonts.ttf import Roboto
import string
-from modules import sd_samplers, shared
+from modules import sd_samplers, shared, script_callbacks
from modules.shared import opts, cmd_opts
LANCZOS = (Image.Resampling.LANCZOS if hasattr(Image, 'Resampling') else Image.LANCZOS)
@@ -539,6 +539,7 @@ def save_image(image, path, basename, seed=None, prompt=None, extension='png', i
else:
txt_fullfn = None
+ script_callbacks.image_saved_callback(image, p, fullfn, txt_fullfn)
return fullfn, txt_fullfn