diff options
author | jovijovi <mageyul@hotmail.com> | 2023-07-06 10:43:17 +0000 |
---|---|---|
committer | jovijovi <mageyul@hotmail.com> | 2023-07-06 10:43:17 +0000 |
commit | 259967b7c60cbd2aeb091e691b5f49d9fb64b872 (patch) | |
tree | c78e5399395e2ba31ee9bc9ff6ff9ee48e292c60 /modules/api/api.py | |
parent | 394ffa7b0a7fff3ec484bcd084e673a8b301ccc8 (diff) | |
download | stable-diffusion-webui-gfx803-259967b7c60cbd2aeb091e691b5f49d9fb64b872.tar.gz stable-diffusion-webui-gfx803-259967b7c60cbd2aeb091e691b5f49d9fb64b872.tar.bz2 stable-diffusion-webui-gfx803-259967b7c60cbd2aeb091e691b5f49d9fb64b872.zip |
fix(api): convert to "RGB" if image mode is "RGBA"
Diffstat (limited to 'modules/api/api.py')
-rw-r--r-- | modules/api/api.py | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/modules/api/api.py b/modules/api/api.py index 2e49526e..6507f641 100644 --- a/modules/api/api.py +++ b/modules/api/api.py @@ -84,6 +84,8 @@ def encode_pil_to_base64(image): image.save(output_bytes, format="PNG", pnginfo=(metadata if use_metadata else None), quality=opts.jpeg_quality) elif opts.samples_format.lower() in ("jpg", "jpeg", "webp"): + if image.mode == "RGBA": + image = image.convert("RGB") parameters = image.info.get('parameters', None) exif_bytes = piexif.dump({ "Exif": { piexif.ExifIFD.UserComment: piexif.helper.UserComment.dump(parameters or "", encoding="unicode") } |