mirrors/stable-diffusion-webui
1
Fork 0
mirror of https://github.com/AUTOMATIC1111/stable-diffusion-webui.git synced 2025-12-24 10:51:03 -08:00
Code Issues Projects Releases Packages Wiki Activity

add webp for embed load

Browse source
This commit is contained in:
DepFA 2022-10-14 18:14:02 +01:00 • committed by AUTOMATIC1111
parent 939f16529a
commit 9a1dcd78ed
1 changed files with 4 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

4
modules/textual_inversion/textual_inversion.py
View file

@ -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")