Merge pull request #7650 from vladmandic/img2img-fix

Convert image from RGBA to RGB before saving
2023-02-19 12:15:27 +03:00 · 2023-02-19 12:15:27 +03:00 · e287d9b294
commit e287d9b294
parent fe46a08f52 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))