diff options
| author | AUTOMATIC <16777216c@gmail.com> | 2023-01-08 05:52:26 +0000 |
|---|---|---|
| committer | AUTOMATIC <16777216c@gmail.com> | 2023-01-08 05:52:26 +0000 |
| commit | a0c87f1fdf2b76b2ae4ef6c4b01ddaede3afab06 (patch) | |
| tree | ebf7590e6b8dbdbfd2b0a0c1a86cdb45bfb8b2ee | |
| parent | 151233399c4b79934bdbb7c12a97eeb6499572fb (diff) | |
| download | stable-diffusion-webui-gfx803-a0c87f1fdf2b76b2ae4ef6c4b01ddaede3afab06.tar.gz stable-diffusion-webui-gfx803-a0c87f1fdf2b76b2ae4ef6c4b01ddaede3afab06.tar.bz2 stable-diffusion-webui-gfx803-a0c87f1fdf2b76b2ae4ef6c4b01ddaede3afab06.zip | |
skip images in embeddings dir if they have a second .preview extension
| -rw-r--r-- | modules/textual_inversion/textual_inversion.py | 4 |
1 files changed, 4 insertions, 0 deletions
diff --git a/modules/textual_inversion/textual_inversion.py b/modules/textual_inversion/textual_inversion.py index 45882ed6..e85dd549 100644 --- a/modules/textual_inversion/textual_inversion.py +++ b/modules/textual_inversion/textual_inversion.py @@ -109,6 +109,10 @@ class EmbeddingDatabase: ext = ext.upper()
if ext in ['.PNG', '.WEBP', '.JXL', '.AVIF']:
+ _, second_ext = os.path.splitext(name)
+ if second_ext.upper() == '.PREVIEW':
+ return
+
embed_image = Image.open(path)
if hasattr(embed_image, 'text') and 'sd-ti-embedding' in embed_image.text:
data = embedding_from_b64(embed_image.text['sd-ti-embedding'])
|