From ee9cf0c81d8961d60429f7d41263d9f565027cd0 Mon Sep 17 00:00:00 2001 From: pharmapsychotic <96542870+pharmapsychotic@users.noreply.github.com> Date: Mon, 6 Mar 2023 12:35:49 -0600 Subject: [PATCH] Fix batch mode with images that have alpha channel Thanks @lync11 ! --- scripts/clip_interrogator_ext.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/scripts/clip_interrogator_ext.py b/scripts/clip_interrogator_ext.py index 9ae03de..b171a01 100644 --- a/scripts/clip_interrogator_ext.py +++ b/scripts/clip_interrogator_ext.py @@ -203,7 +203,7 @@ def batch_tab(): for file in files: if shared.state.interrupted: break - image = Image.open(os.path.join(folder, file)) + image = Image.open(os.path.join(folder, file)).convert('RGB') captions.append(ci.generate_caption(image)) shared.total_tqdm.update() @@ -214,7 +214,7 @@ def batch_tab(): for idx, file in enumerate(files): if shared.state.interrupted: break - image = Image.open(os.path.join(folder, file)) + image = Image.open(os.path.join(folder, file)).convert('RGB') prompt = interrogate(image, mode, caption=captions[idx]) writer.add(file, prompt) shared.total_tqdm.update()