diff options
| author | AUTOMATIC1111 <16777216c@gmail.com> | 2022-10-11 14:25:36 +0000 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2022-10-11 14:25:36 +0000 |
| commit | 4f96ffd0b5b5fadbaac5d08ce1d57384646ababa (patch) | |
| tree | 88640250ec4e8ad5dd8cc47612e4237264c79b04 /modules/textual_inversion/preprocess.py | |
| parent | 861db783c7acfcb93cf0b5191db3d50f9a9bc531 (diff) | |
| parent | 8bacbca0a1ab9aabcb0ad0cbf070e0006991e98a (diff) | |
| download | stable-diffusion-webui-gfx803-4f96ffd0b5b5fadbaac5d08ce1d57384646ababa.tar.gz stable-diffusion-webui-gfx803-4f96ffd0b5b5fadbaac5d08ce1d57384646ababa.tar.bz2 stable-diffusion-webui-gfx803-4f96ffd0b5b5fadbaac5d08ce1d57384646ababa.zip | |
Merge pull request #2201 from alg-wiki/textual__inversion
Textual Inversion: Preprocess and Training will only pick-up image files instead
Diffstat (limited to 'modules/textual_inversion/preprocess.py')
| -rw-r--r-- | modules/textual_inversion/preprocess.py | 5 |
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
|