add webp for embed load

9a1dcd78 · DepFA · AUTOMATIC1111 · 939f1652 · 9a1dcd78
Commit 9a1dcd78 authored Oct 14, 2022 by DepFA Committed by AUTOMATIC1111 Oct 15, 2022
Hide whitespace changes
Inline Side-by-side

Showing with 4 additions and 0 deletions

modules/textual_inversion/textual_inversion.py modules/textual_inversion/textual_inversion.py +4 -0

No files found.
--- 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")