aboutsummaryrefslogtreecommitdiffstats
path: root/modules/images.py
diff options
context:
space:
mode:
authorVladimir Mandic <mandic00@live.com>2022-12-26 14:53:26 +0000
committerVladimir Mandic <mandic00@live.com>2022-12-26 14:56:19 +0000
commitae955b0146a52ea2474c79655ede0d361829ef63 (patch)
tree5f47d09e8770fd57bd9ca740b42363cb78ee98e8 /modules/images.py
parent4af3ca5393151d61363c30eef4965e694eeac15e (diff)
downloadstable-diffusion-webui-gfx803-ae955b0146a52ea2474c79655ede0d361829ef63.tar.gz
stable-diffusion-webui-gfx803-ae955b0146a52ea2474c79655ede0d361829ef63.tar.bz2
stable-diffusion-webui-gfx803-ae955b0146a52ea2474c79655ede0d361829ef63.zip
fix rgba to rgb when using jpeg output
Diffstat (limited to 'modules/images.py')
-rw-r--r--modules/images.py3
1 files changed, 3 insertions, 0 deletions
diff --git a/modules/images.py b/modules/images.py
index 31d4528d..962a955d 100644
--- a/modules/images.py
+++ b/modules/images.py
@@ -525,6 +525,9 @@ def save_image(image, path, basename, seed=None, prompt=None, extension='png', i
image_to_save.save(temp_file_path, format=image_format, quality=opts.jpeg_quality, pnginfo=pnginfo_data)
elif extension.lower() in (".jpg", ".jpeg", ".webp"):
+ if image_to_save.mode == 'RGBA':
+ image_to_save = image_to_save.convert("RGB")
+
image_to_save.save(temp_file_path, format=image_format, quality=opts.jpeg_quality)
if opts.enable_pnginfo and info is not None: