revi13 commited on
Commit
29b0cd6
·
verified ·
1 Parent(s): 4f57fbc

Update ip_adapter/ip_adapter_faceid.py

Browse files
Files changed (1) hide show
  1. ip_adapter/ip_adapter_faceid.py +2 -2
ip_adapter/ip_adapter_faceid.py CHANGED
@@ -175,9 +175,9 @@ class IPAdapterFaceID:
175
  state_dict["ip_adapter"][key.replace("ip_adapter.", "")] = f.get_tensor(key)
176
  else:
177
  state_dict = torch.load(self.ip_ckpt, map_location="cpu")
178
- self.image_proj_model.load_state_dict(state_dict["image_proj"])
179
  ip_layers = torch.nn.ModuleList(self.pipe.unet.attn_processors.values())
180
- ip_layers.load_state_dict(state_dict["ip_adapter"])
181
 
182
  @torch.inference_mode()
183
  def get_image_embeds(self, faceid_embeds):
 
175
  state_dict["ip_adapter"][key.replace("ip_adapter.", "")] = f.get_tensor(key)
176
  else:
177
  state_dict = torch.load(self.ip_ckpt, map_location="cpu")
178
+ self.image_proj_model.load_state_dict(state_dict["image_proj"], strict=False)
179
  ip_layers = torch.nn.ModuleList(self.pipe.unet.attn_processors.values())
180
+ ip_layers.load_state_dict(state_dict["ip_adapter"], strict=False)
181
 
182
  @torch.inference_mode()
183
  def get_image_embeds(self, faceid_embeds):