Re-register embed load callback after model load if needed

2024-08-11 14:55:35 +02:00 · 2024-08-11 14:55:35 +02:00 · e2cfe7341b
parent ce51ec52a2
commit e2cfe7341b
1 changed files with 3 additions and 0 deletions
--- a/scripts/tag_autocomplete_helper.py
+++ b/scripts/tag_autocomplete_helper.py
@ -214,6 +214,9 @@ def get_embeddings(sd_model):

    try:
        embed_db = get_embed_db(sd_model)
+        # Re-register callback if needed
+        if embed_db is not None and load_textual_inversion_embeddings != embed_db.load_textual_inversion_embeddings:
+            load_textual_inversion_embeddings = embed_db.load_textual_inversion_embeddings
        
        loaded = embed_db.word_embeddings
        skipped = embed_db.skipped_embeddings