diff options
| author | DepFA <35278260+dfaker@users.noreply.github.com> | 2022-10-14 18:23:20 +0100 |
|---|---|---|
| committer | AUTOMATIC1111 <16777216c@gmail.com> | 2022-10-15 17:23:39 +0300 |
| commit | ddf6899df0cf87d4da77cb2ce223061f4a5edf18 (patch) | |
| tree | d09a8e28fe4e84201ea752bfff0c0a746005914d /modules/textual_inversion | |
| parent | 9a1dcd78edbf9caf68b9e6286d7b5ca81500e243 (diff) | |
generalise to popular lossless formats
Diffstat (limited to 'modules/textual_inversion')
| -rw-r--r-- | modules/textual_inversion/textual_inversion.py | 8 |
1 files changed, 2 insertions, 6 deletions
diff --git a/modules/textual_inversion/textual_inversion.py b/modules/textual_inversion/textual_inversion.py index c07bffc3..b99df3b1 100644 --- a/modules/textual_inversion/textual_inversion.py +++ b/modules/textual_inversion/textual_inversion.py @@ -88,18 +88,14 @@ class EmbeddingDatabase: data = []
- if filename.upper().endswith('.PNG'):
+ if os.path.splitext(filename.upper())[-1] in ['.PNG', '.WEBP', '.JXL', '.AVIF']:
embed_image = Image.open(path)
- if 'sd-ti-embedding' in embed_image.text:
+ if hasattr(embed_image, 'text') and 'sd-ti-embedding' in embed_image.text:
data = embedding_from_b64(embed_image.text['sd-ti-embedding'])
name = data.get('name', name)
else:
data = extract_image_data_embed(embed_image)
name = data.get('name', name)
- elif filename.upper().endswith('.WEBP'):
- embed_image = Image.open(path)
- data = extract_image_data_embed(embed_image)
- name = data.get('name', name)
else:
data = torch.load(path, map_location="cpu")
|