DominikDoom/a1111-sd-webui-tagcomplete
1
0
Fork 0
mirror of https://github.com/DominikDoom/a1111-sd-webui-tagcomplete.git synced 2026-03-13 01:00:08 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix SD.Next error caused by embeddings without filenames

Browse Source 
This only ignores these embeddings, the root cause is a bug / behavioral difference in SD.Next
Fixes #242
This commit is contained in:
DominikDoom
2023-09-26 10:27:50 +02:00
parent 1a9157fe6e
commit 94ec8884c3
1 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
scripts/tag_autocomplete_helper.py
View File

@@ -174,14 +174,14 @@ def get_embeddings(sd_model):
# Add embeddings to the correct list
if (emb_a_shape == V1_SHAPE):
emb_v1 = [(Path(v.filename), k, "v1") for (k,v) in emb_type_a.items()]
emb_v1 = [(Path(v.filename), k, "v1") for (k,v) in emb_type_a.items() if v.filename is not None]
elif (emb_a_shape == V2_SHAPE):
emb_v2 = [(Path(v.filename), k, "v2") for (k,v) in emb_type_a.items()]
emb_v2 = [(Path(v.filename), k, "v2") for (k,v) in emb_type_a.items() if v.filename is not None]
if (emb_b_shape == V1_SHAPE):
emb_v1 = [(Path(v.filename), k, "v1") for (k,v) in emb_type_b.items()]
emb_v1 = [(Path(v.filename), k, "v1") for (k,v) in emb_type_b.items() if v.filename is not None]
elif (emb_b_shape == V2_SHAPE):
emb_v2 = [(Path(v.filename), k, "v2") for (k,v) in emb_type_b.items()]
emb_v2 = [(Path(v.filename), k, "v2") for (k,v) in emb_type_b.items() if v.filename is not None]
# Get shape of current model
#vec = sd_model.cond_stage_model.encode_embedding_init_text(",", 1)