aboutsummaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorAUTOMATIC <16777216c@gmail.com>2023-01-03 18:39:14 +0300
committerAUTOMATIC <16777216c@gmail.com>2023-01-03 18:39:14 +0300
commit8f96f9289981a66741ba770d14f3d27ce335a0fb (patch)
treee6a8832d705ba417b6ecfdcb677b62dca7ffc1c9
parent2d5a5076bb2a0c05cc27d75a1bcadab7f32a46d0 (diff)
call script callbacks for reloaded model after loading embeddings
-rw-r--r--modules/sd_models.py4
1 files changed, 2 insertions, 2 deletions
diff --git a/modules/sd_models.py b/modules/sd_models.py
index bff8d6c9..b98b05fc 100644
--- a/modules/sd_models.py
+++ b/modules/sd_models.py
@@ -324,12 +324,12 @@ def load_model(checkpoint_info=None):
sd_model.eval()
shared.sd_model = sd_model
+ 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
+
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