Safety check for missing shape

2023-09-26 15:12:29 +02:00 · 2023-09-26 15:12:29 +02:00 · b3e71e840d
parent 998514bebb
commit b3e71e840d
1 changed files with 1 additions and 1 deletions
--- a/scripts/tag_autocomplete_helper.py
+++ b/scripts/tag_autocomplete_helper.py
@ -169,7 +169,7 @@ def get_embeddings(sd_model):

        # Add embeddings to the correct list
        for key, emb in (loaded | skipped).items():
-            if (emb.filename is None):
+            if emb.filename is None or emb.shape is None:
                continue

            if emb.shape == V1_SHAPE: