Mirrored_Repos/InvokeAI
1
0
Fork 0
mirror of https://github.com/invoke-ai/InvokeAI synced 2024-08-30 20:32:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix: Fail when unexpected keys are found in IP Adapter models

Browse Source
This commit is contained in:
blessedcoolant 2024-03-29 12:34:56 +05:30
parent 56ed697c23
commit 6e4c2d3685
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
invokeai/backend/ip_adapter/ip_adapter.py
View File

@ -214,6 +214,8 @@ def load_ip_adapter_tensors(ip_adapter_ckpt_path: str, device: str) -> IPAdapter
state_dict["image_proj"][key.replace("image_proj.", "")] = model.get_tensor(key) state_dict["image_proj"][key.replace("image_proj.", "")] = model.get_tensor(key)
elif key.startswith("ip_adapter."): elif key.startswith("ip_adapter."):
state_dict["ip_adapter"][key.replace("ip_adapter.", "")] = model.get_tensor(key) state_dict["ip_adapter"][key.replace("ip_adapter.", "")] = model.get_tensor(key)
else:
raise RuntimeError(f"Encountered unexpected IP Adapter state dict key: '{key}'.")
else: else:
ip_adapter_diffusers_checkpoint_path = ip_adapter_ckpt_path + "/ip_adapter.bin" ip_adapter_diffusers_checkpoint_path = ip_adapter_ckpt_path + "/ip_adapter.bin"
state_dict = torch.load(ip_adapter_diffusers_checkpoint_path, map_location="cpu") state_dict = torch.load(ip_adapter_diffusers_checkpoint_path, map_location="cpu")