convert rgba to rgb

some image format (e.g. jpg) do not support rgba
3 years ago · 4c562a9832
parent ea9bd9fc74
commit 4c562a9832
1 changed files with 2 additions and 0 deletions
--- a/modules/img2img.py
+++ b/modules/img2img.py
@ -73,6 +73,8 @@ def process_batch(p, input_dir, output_dir, inpaint_mask_dir, args):

            if not save_normally:
                os.makedirs(output_dir, exist_ok=True)
+                if processed_image.mode == 'RGBA':
+                    processed_image = processed_image.convert("RGB")
                processed_image.save(os.path.join(output_dir, filename))