diff options
| author | Trung Ngo <codem01@gmail.com> | 2022-10-24 06:17:09 +0000 |
|---|---|---|
| committer | Trung Ngo <codem01@gmail.com> | 2022-10-24 06:25:31 +0000 |
| commit | 734986dde3231416813f827242c111da212b2ccb (patch) | |
| tree | 85e3ae4a7e079ee34b4e4220baad222ca4319d41 /modules/images.py | |
| parent | e9a410b5357612f63528015c5533c2185dcff92e (diff) | |
| download | stable-diffusion-webui-gfx803-734986dde3231416813f827242c111da212b2ccb.tar.gz stable-diffusion-webui-gfx803-734986dde3231416813f827242c111da212b2ccb.tar.bz2 stable-diffusion-webui-gfx803-734986dde3231416813f827242c111da212b2ccb.zip | |
add callback after image is saved
Diffstat (limited to 'modules/images.py')
| -rw-r--r-- | modules/images.py | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/modules/images.py b/modules/images.py index b9589563..01c60f89 100644 --- a/modules/images.py +++ b/modules/images.py @@ -12,7 +12,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)
@@ -467,6 +467,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
|