Merge branch 'master' into beta

jn-jairo · Nov 8, 2023 · d47357f · d47357f
2 parents f8f3f94 + 0a6fd49
commit d47357f
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 0 deletions.
diff --git a/comfy/model_patcher.py b/comfy/model_patcher.py
@@ -40,6 +40,7 @@ def clone(self):
         for k in self.patches:
             n.patches[k] = self.patches[k][:]
 
+        n.object_patches = self.object_patches.copy()
         n.model_options = copy.deepcopy(self.model_options)
         n.model_keys = self.model_keys
         return n

diff --git a/comfy/sd.py b/comfy/sd.py
@@ -513,6 +513,9 @@ def load_unet(unet_path): #load unet in diffusers format
     model = model_config.get_model(new_sd, "")
     model = model.to(offload_device)
     model.load_model_weights(new_sd, "")
+    left_over = sd.keys()
+    if len(left_over) > 0:
+        print("left over keys in unet:", left_over)
     return comfy.model_patcher.ModelPatcher(model, load_device=model_management.get_torch_device(), offload_device=offload_device)
 
 def save_checkpoint(output_path, model, clip, vae, metadata=None):