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

Embedding forced reload.

Browse Source
This commit is contained in:
Symbiomatrix
2023-08-20 18:51:37 +03:00
committed by GitHub
parent 651cf5fb46
commit 5b69d1e622
1 changed files with 8 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
scripts/tag_autocomplete_helper.py
View File

@@ -17,6 +17,12 @@ from scripts.model_keyword_support import (get_lora_simple_hash,
write_model_keyword_path)
from scripts.shared_paths import *
# Attempt to get embedding load function, using the same call as api.
try:
load_textual_inversion_embeddings = sd_hijack.model_hijack.embedding_db.load_textual_inversion_embeddings
except Exception as e: # Not supported.
load_textual_inversion_embeddings = lambda *args, **kwargs: None
print("Cannot load embeddings instantly: ", e)
def get_wildcards():
"""Returns a list of all wildcards. Works on nested folders."""
@@ -280,6 +286,7 @@ if EMB_PATH.exists():
def refresh_temp_files():
global WILDCARD_EXT_PATHS
WILDCARD_EXT_PATHS = find_ext_wildcard_paths()
load_textual_inversion_embeddings(force_reload = True) # Instant embedding reload.
write_temp_files()
get_embeddings(shared.sd_model)
@@ -505,4 +512,4 @@ def api_tac(_: gr.Blocks, app: FastAPI):
script_callbacks.on_app_started(api_tac)
script_callbacks.on_app_started(api_tac)