nsfwalex commited on
Commit
9555f5e
1 Parent(s): c7f4bd1

Update ipown.py

Browse files
Files changed (1) hide show
  1. ipown.py +3 -0
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