Commit 9a1dcd78 authored by DepFA's avatar DepFA Committed by AUTOMATIC1111
Browse files

add webp for embed load

parent 939f1652

modules/textual_inversion/textual_inversion.py

+4 −0
Original line number Diff line number Diff line
@@ -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")

CRA Git | Maintained and supported by SUSTech CRA and CCSE