fix for broken checkpoint merger

3 years ago · f894dd552f
parent 9931c0bd48
commit f894dd552f
1 changed files with 4 additions and 1 deletions
--- a/modules/sd_models.py
+++ b/modules/sd_models.py
@ -148,7 +148,10 @@ def get_state_dict_from_checkpoint(pl_sd):
        if new_key is not None:
            sd[new_key] = v

-    return sd
+    pl_sd.clear()
+    pl_sd.update(sd)
+
+    return pl_sd


 def load_model_weights(model, checkpoint_info):