michael/stable-diffusion-webui
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

fix using aaaa-100 embedding when the prompt has aaaa-10000 and you have both aaaa-100 and aaaa-10000 in the directory with embeddings.

Browse source
This commit is contained in:
AUTOMATIC 2022-10-02 19:56:37 +03:00
parent 88ec0cf557
commit 71fe7fa49f
1 changed files with 2 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
modules/textual_inversion/textual_inversion.py
View file

@ -57,7 +57,8 @@ class EmbeddingDatabase:
first_id = ids[0] first_id = ids[0]
if first_id not in self.ids_lookup: if first_id not in self.ids_lookup:
self.ids_lookup[first_id] = [] self.ids_lookup[first_id] = []
self.ids_lookup[first_id].append((ids, embedding))
self.ids_lookup[first_id] = sorted(self.ids_lookup[first_id] + [(ids, embedding)], key=lambda x: len(x[0]), reverse=True)
return embedding return embedding