fix(api): convert to "RGB" if image mode is "RGBA"

This commit is contained in:
jovijovi 2023-07-06 18:43:17 +08:00
parent 394ffa7b0a
commit 259967b7c6
1 changed files with 2 additions and 0 deletions

View File

@ -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") }