diff options
| author | AUTOMATIC <16777216c@gmail.com> | 2023-01-03 15:39:14 +0000 |
|---|---|---|
| committer | AUTOMATIC <16777216c@gmail.com> | 2023-01-03 15:39:14 +0000 |
| commit | 8f96f9289981a66741ba770d14f3d27ce335a0fb (patch) | |
| tree | e6a8832d705ba417b6ecfdcb677b62dca7ffc1c9 | |
| parent | 2d5a5076bb2a0c05cc27d75a1bcadab7f32a46d0 (diff) | |
| download | stable-diffusion-webui-gfx803-8f96f9289981a66741ba770d14f3d27ce335a0fb.tar.gz stable-diffusion-webui-gfx803-8f96f9289981a66741ba770d14f3d27ce335a0fb.tar.bz2 stable-diffusion-webui-gfx803-8f96f9289981a66741ba770d14f3d27ce335a0fb.zip | |
call script callbacks for reloaded model after loading embeddings
| -rw-r--r-- | modules/sd_models.py | 4 |
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
|