Spaces:
Sleeping
Sleeping
Update ip_adapter/ip_adapter_faceid.py
Browse files
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):
|