add webp for embed load

2022-10-14 18:14:02 +01:00
parent 939f16529a
commit 9a1dcd78ed
1 changed files with 4 additions and 0 deletions
@@ -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")