diff options
| author | brkirch <brkirch@users.noreply.github.com> | 2023-01-04 00:40:16 -0500 |
|---|---|---|
| committer | brkirch <brkirch@users.noreply.github.com> | 2023-01-06 00:14:20 -0500 |
| commit | f6ab5a39d762a7791573d1c52ae5a3024b10e8ed (patch) | |
| tree | c3958d77a6dae42457b571dbe0f1efec7ce45dd2 /modules/sd_models.py | |
| parent | d782a95967c9eea753df3333cd1954b6ec73eba0 (diff) | |
| parent | 3e22e294135ed0327ce9d9738655ff03c53df3c0 (diff) | |
Merge branch 'AUTOMATIC1111:master' into sub-quad_attn_opt
Diffstat (limited to 'modules/sd_models.py')
| -rw-r--r-- | modules/sd_models.py | 5 |
1 files changed, 5 insertions, 0 deletions
diff --git a/modules/sd_models.py b/modules/sd_models.py index ecdd91c5..b98b05fc 100644 --- a/modules/sd_models.py +++ b/modules/sd_models.py @@ -228,6 +228,8 @@ def load_model_weights(model, checkpoint_info, vae_file="auto"): model.sd_model_checkpoint = checkpoint_file
model.sd_checkpoint_info = checkpoint_info
+ model.logvar = model.logvar.to(devices.device) # fix for training
+
sd_vae.delete_base_vae()
sd_vae.clear_loaded_vae()
vae_file = sd_vae.resolve_vae(checkpoint_file, vae_file=vae_file)
@@ -322,9 +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.")
+
return sd_model
|