aboutsummaryrefslogtreecommitdiff
path: root/modules/textual_inversion/preprocess.py
diff options
context:
space:
mode:
authorAUTOMATIC1111 <16777216c@gmail.com>2022-10-11 21:50:19 +0300
committerGitHub <noreply@github.com>2022-10-11 21:50:19 +0300
commit419e539fe33e5bd1f7c7f55b477f955f6bdfc822 (patch)
tree131ea9501cefd29b7e5cb4b3e1ea36d76c64beb6 /modules/textual_inversion/preprocess.py
parent2536ecbb1790da2af0d61b6a26f38732cba665cd (diff)
parentd7474a5185df2af84a93a12bc7e140d24e0fc516 (diff)
Merge branch 'learning_rate-scheduling' into learnschedule
Diffstat (limited to 'modules/textual_inversion/preprocess.py')
-rw-r--r--modules/textual_inversion/preprocess.py5
1 files changed, 4 insertions, 1 deletions
diff --git a/modules/textual_inversion/preprocess.py b/modules/textual_inversion/preprocess.py
index d7efdef2..1a672725 100644
--- a/modules/textual_inversion/preprocess.py
+++ b/modules/textual_inversion/preprocess.py
@@ -46,7 +46,10 @@ def preprocess(process_src, process_dst, process_width, process_height, process_
for index, imagefile in enumerate(tqdm.tqdm(files)):
subindex = [0]
filename = os.path.join(src, imagefile)
- img = Image.open(filename).convert("RGB")
+ try:
+ img = Image.open(filename).convert("RGB")
+ except Exception:
+ continue
if shared.state.interrupted:
break