OPENGIT/stable-diffusion-webui
1
0
Fork 0
mirror of https://github.com/AUTOMATIC1111/stable-diffusion-webui.git synced 2025-01-19 21:00:14 +08:00
Code Issues Packages Projects Releases Wiki Activity

FIX: initial_pp RGBA right way

Browse Source
This commit is contained in:
Art Gourieff 2024-03-20 15:14:28 +07:00
parent 61f488302f
commit 702edb288e
2 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
modules/api/api.py
View File

@ -99,7 +99,7 @@ def decode_base64_to_image(encoding):
raise HTTPException(status_code=500, detail="Invalid encoded image") from e
def encode_pil_to_base64(image):
def encode_pil_to_base64(image: Image.Image):
with io.BytesIO() as output_bytes:
if isinstance(image, str):
return image

2
modules/postprocessing.py
View File

@ -66,7 +66,7 @@ def run_postprocessing(extras_mode, image, image_folder, input_dir, output_dir,
if parameters:
existing_pnginfo["parameters"] = parameters
initial_pp = scripts_postprocessing.PostprocessedImage(image_data)
initial_pp = scripts_postprocessing.PostprocessedImage(image_data.convert("RGBA")) if image_data.mode == "RGBA" else scripts_postprocessing.PostprocessedImage(image_data.convert("RGB"))
scripts.scripts_postproc.run(initial_pp, args)