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

Re-register embed load callback after model load if needed

Browse Source
This commit is contained in:
DominikDoom
2024-08-11 14:55:35 +02:00
parent ce51ec52a2
commit e2cfe7341b
1 changed files with 3 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
scripts/tag_autocomplete_helper.py
View File

@@ -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