diff --git a/modules/sd_models.py b/modules/sd_models.py
index 7ad6d474..eae22e87 100644
--- 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):