diff options
| author | Keith <1868690+wk5ovc@users.noreply.github.com> | 2023-05-15 02:47:01 +0000 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2023-05-15 02:47:01 +0000 |
| commit | f517838c75014f981ae1c41f1bc776d74daf9a23 (patch) | |
| tree | eb2bece819d155bf937fba021831b3cfe9a977dc /modules/ui_extra_networks.py | |
| parent | 89f9faa63388756314e8a1d96cf86bf5e0663045 (diff) | |
| download | stable-diffusion-webui-gfx803-f517838c75014f981ae1c41f1bc776d74daf9a23.tar.gz stable-diffusion-webui-gfx803-f517838c75014f981ae1c41f1bc776d74daf9a23.tar.bz2 stable-diffusion-webui-gfx803-f517838c75014f981ae1c41f1bc776d74daf9a23.zip | |
Fix extra networks save preview image geninfo
Diffstat (limited to 'modules/ui_extra_networks.py')
| -rw-r--r-- | modules/ui_extra_networks.py | 16 |
1 files changed, 13 insertions, 3 deletions
diff --git a/modules/ui_extra_networks.py b/modules/ui_extra_networks.py index 8c3dea56..9e6e0531 100644 --- a/modules/ui_extra_networks.py +++ b/modules/ui_extra_networks.py @@ -334,9 +334,19 @@ def setup_ui(ui, gallery): assert is_allowed, f'writing to {filename} is not allowed'
if geninfo:
- pnginfo_data = PngImagePlugin.PngInfo()
- pnginfo_data.add_text('parameters', geninfo)
- image.save(filename, pnginfo=pnginfo_data)
+ ext = os.path.splitext(filename)[1].lower()
+ if ext == '.png':
+ pnginfo_data = PngImagePlugin.PngInfo()
+ pnginfo_data.add_text('parameters', geninfo)
+ image.save(filename, pnginfo=pnginfo_data)
+ elif ext in ('.jpg', '.jpeg', '.webp'):
+ exif_bytes = piexif.dump({
+ 'Exif': {piexif.ExifIFD.UserComment: piexif.helper.UserComment.dump(geninfo or '',
+ encoding='unicode')}
+ })
+ image.save(filename, exif=exif_bytes, quality=shared.opts.jpeg_quality)
+ else:
+ image.save(filename)
else:
image.save(filename)
|