Update ipown.py
Browse files
ipown.py
CHANGED
|
@@ -183,6 +183,9 @@ class IPAdapterFaceID:
|
|
| 183 |
def get_image_embeds(self, faceid_embeds):
|
| 184 |
|
| 185 |
faceid_embeds = faceid_embeds.to(self.device, dtype=self.torch_dtype)
|
|
|
|
|
|
|
|
|
|
| 186 |
image_prompt_embeds = self.image_proj_model(faceid_embeds)
|
| 187 |
uncond_image_prompt_embeds = self.image_proj_model(torch.zeros_like(faceid_embeds))
|
| 188 |
return image_prompt_embeds, uncond_image_prompt_embeds
|
|
|
|
| 183 |
def get_image_embeds(self, faceid_embeds):
|
| 184 |
|
| 185 |
faceid_embeds = faceid_embeds.to(self.device, dtype=self.torch_dtype)
|
| 186 |
+
print(faceid_embeds.device)
|
| 187 |
+
print(next(self.image_proj_model.parameters()).device)
|
| 188 |
+
|
| 189 |
image_prompt_embeds = self.image_proj_model(faceid_embeds)
|
| 190 |
uncond_image_prompt_embeds = self.image_proj_model(torch.zeros_like(faceid_embeds))
|
| 191 |
return image_prompt_embeds, uncond_image_prompt_embeds
|