fix for incorrect model weight loading for #814

1 files changed, 5 insertions, 1 deletions

diff --git a/modules/sd_models.py b/modules/sd_models.py
index 7a5edced..eb21e498 100644
--- a/modules/sd_models.py
+++ b/modules/sd_models.py
@@ -137,7 +137,7 @@ def load_model():
 

 

 def reload_model_weights(sd_model, info=None):

-    from modules import lowvram, devices

+    from modules import lowvram, devices, sd_hijack

     checkpoint_info = info or select_checkpoint()

 

     if sd_model.sd_model_checkpint == checkpoint_info.filename:

@@ -148,8 +148,12 @@ def reload_model_weights(sd_model, info=None):
     else:

         sd_model.to(devices.cpu)

 

+    sd_hijack.model_hijack.undo_hijack(sd_model)

+

     load_model_weights(sd_model, checkpoint_info.filename, checkpoint_info.hash)

 

+    sd_hijack.model_hijack.hijack(sd_model)

+

     if not shared.cmd_opts.lowvram and not shared.cmd_opts.medvram:

         sd_model.to(devices.device)