pharmapsychotic/clip-interrogator-ext
1
0
Fork 0
mirror of https://github.com/pharmapsychotic/clip-interrogator-ext.git synced 2026-04-28 10:11:27 +00:00
Code Issues Packages Projects Releases Wiki Activity

remove IndexError handler

Browse Source
This commit is contained in:
genevera
2023-06-01 00:38:29 -04:00
parent a65b4fbb6c
commit be5a183ae9
1 changed files with 9 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
scripts/clip_interrogator_ext.py
View File

@@ -209,11 +209,15 @@ def batch_tab():
if shared.state.interrupted:
break
image = Image.open(os.path.join(folder, file)).convert('RGB')
captions.append(ci.generate_caption(image))
except OSError:
print(f" Could not read {file}; continuing")
caption = ci.generate_caption(image)
shared.total_tqdm.update()
except OSError as e:
print(f"{e}; continuing")
# https://github.com/pharmapsychotic/clip-interrogator-ext/pull/49#pullrequestreview-1454127223
caption = ""
continue
finally:
captions.append(caption)
shared.total_tqdm.update()
# interrogate in second pass
@@ -225,12 +229,8 @@ def batch_tab():
if shared.state.interrupted:
break
image = Image.open(os.path.join(folder, file)).convert('RGB')
try:
prompt = interrogate(image, mode, caption=captions[idx])
writer.add(file, prompt)
except IndexError as e:
print(f" {e}, continuing")
continue
prompt = interrogate(image, mode, caption=captions[idx])
writer.add(file, prompt)
except OSError as e:
print(f" {e}, continuing")
continue