pharmapsychotic/clip-interrogator-ext
1
0
Fork 0
mirror of https://github.com/pharmapsychotic/clip-interrogator-ext.git synced 2026-03-13 17:29:59 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix batch mode with images that have alpha channel

Browse Source 
Thanks @lync11 !
This commit is contained in:
pharmapsychotic
2023-03-06 12:35:49 -06:00
committed by GitHub
parent 6385cd862f
commit ee9cf0c81d
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
scripts/clip_interrogator_ext.py
View File

@@ -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()