diff options
| author | AUTOMATIC <16777216c@gmail.com> | 2022-10-02 19:56:37 +0300 |
|---|---|---|
| committer | AUTOMATIC <16777216c@gmail.com> | 2022-10-02 19:56:37 +0300 |
| commit | 71fe7fa49f5eb1a2c89932a9d217ed153c12fc8b (patch) | |
| tree | 4bdf2acc122a8c52e3d63ffac801188af3271d47 /modules/textual_inversion/textual_inversion.py | |
| parent | 88ec0cf5571883d84abd09196652b3679e359f2e (diff) | |
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.
Diffstat (limited to 'modules/textual_inversion/textual_inversion.py')
| -rw-r--r-- | modules/textual_inversion/textual_inversion.py | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/modules/textual_inversion/textual_inversion.py b/modules/textual_inversion/textual_inversion.py index 0c50161d..9d2241ce 100644 --- a/modules/textual_inversion/textual_inversion.py +++ b/modules/textual_inversion/textual_inversion.py @@ -57,7 +57,8 @@ class EmbeddingDatabase: first_id = ids[0]
if first_id not in self.ids_lookup:
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
|