aboutsummaryrefslogtreecommitdiff
path: root/modules/sd_models.py
diff options
context:
space:
mode:
authorVladimir Mandic <mandic00@live.com>2022-12-31 11:27:02 -0500
committerGitHub <noreply@github.com>2022-12-31 11:27:02 -0500
commitf55ac33d446185680604e872ceda2ae858821d5c (patch)
tree9189d172b955e1d135b386018df67d8e0eee12fd /modules/sd_models.py
parentf34c7341720fb2059992926c9f9ae6ff25f7385b (diff)
validate textual inversion embeddings
Diffstat (limited to 'modules/sd_models.py')
-rw-r--r--modules/sd_models.py3
1 files changed, 3 insertions, 0 deletions
diff --git a/modules/sd_models.py b/modules/sd_models.py
index ecdd91c5..ebd4dff7 100644
--- a/modules/sd_models.py
+++ b/modules/sd_models.py
@@ -325,6 +325,9 @@ def load_model(checkpoint_info=None):
script_callbacks.model_loaded_callback(sd_model)
print("Model loaded.")
+
+ sd_hijack.model_hijack.embedding_db.load_textual_inversion_embeddings(force_reload = True) # Reload embeddings after model load as they may or may not fit the model
+
return sd_model