diff options
| author | DepFA <35278260+dfaker@users.noreply.github.com> | 2022-10-14 17:14:02 +0000 |
|---|---|---|
| committer | AUTOMATIC1111 <16777216c@gmail.com> | 2022-10-15 14:23:39 +0000 |
| commit | 9a1dcd78edbf9caf68b9e6286d7b5ca81500e243 (patch) | |
| tree | 88191ca7fa032b6038e96d06dcf1b2790bb2dbac /modules/textual_inversion/textual_inversion.py | |
| parent | 939f16529a72fe48c2ce3ef31bdaba785925a33c (diff) | |
| download | stable-diffusion-webui-gfx803-9a1dcd78edbf9caf68b9e6286d7b5ca81500e243.tar.gz stable-diffusion-webui-gfx803-9a1dcd78edbf9caf68b9e6286d7b5ca81500e243.tar.bz2 stable-diffusion-webui-gfx803-9a1dcd78edbf9caf68b9e6286d7b5ca81500e243.zip | |
add webp for embed load
Diffstat (limited to 'modules/textual_inversion/textual_inversion.py')
| -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 1d697c90..c07bffc3 100644 --- a/modules/textual_inversion/textual_inversion.py +++ b/modules/textual_inversion/textual_inversion.py @@ -96,6 +96,10 @@ class EmbeddingDatabase: 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")
|