Skip to content

GitLab

"ppocr/git@developer.sourcefind.cn:wangsen/paddle_dbnet.git" did not exist on "d6ee6bdb487b34d21c77f7d47522086c9ddfefaf"
Commit 0a6fd49a authored by comfyanonymous's avatar comfyanonymous
Browse files

Print leftover keys when using the UNETLoader.

parent fe40109b
Hide whitespace changes
Inline Side-by-side
Showing with 3 additions and 0 deletions
comfy/sd.py
View file @ 0a6fd49a
...@@ -496,6 +496,9 @@ def load_unet(unet_path): #load unet in diffusers format ...@@ -496,6 +496,9 @@ def load_unet(unet_path): #load unet in diffusers format
model = model_config.get_model(new_sd, "") model = model_config.get_model(new_sd, "")
model = model.to(offload_device) model = model.to(offload_device)
model.load_model_weights(new_sd, "") 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) 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): def save_checkpoint(output_path, model, clip, vae, metadata=None):
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment