Commit a0c87f1f authored by AUTOMATIC's avatar AUTOMATIC
Browse files

skip images in embeddings dir if they have a second .preview extension

parent 15123339

modules/textual_inversion/textual_inversion.py

+4 −0
Original line number Diff line number Diff line
@@ -109,6 +109,10 @@ class EmbeddingDatabase: 
            ext = ext.upper()



            if ext in ['.PNG', '.WEBP', '.JXL', '.AVIF']:

                _, second_ext = os.path.splitext(name)

                if second_ext.upper() == '.PREVIEW':

                    return



                embed_image = Image.open(path)

                if hasattr(embed_image, 'text') and 'sd-ti-embedding' in embed_image.text:

                    data = embedding_from_b64(embed_image.text['sd-ti-embedding'])

CRA Git | Maintained and supported by SUSTech CRA and CCSE