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
|