From e2cfe7341b629252aec8e5607fdaf1f1e446de41 Mon Sep 17 00:00:00 2001 From: DominikDoom Date: Sun, 11 Aug 2024 14:55:35 +0200 Subject: [PATCH] Re-register embed load callback after model load if needed --- scripts/tag_autocomplete_helper.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/scripts/tag_autocomplete_helper.py b/scripts/tag_autocomplete_helper.py index 3d8eb00..7579264 100644 --- 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