revi13 commited on
Commit
6ed6e37
·
verified ·
1 Parent(s): 74415da

Update ip_adapter/ip_adapter_faceid.py

Browse files
Files changed (1) hide show
  1. ip_adapter/ip_adapter_faceid.py +3 -0
ip_adapter/ip_adapter_faceid.py CHANGED
@@ -515,6 +515,9 @@ class IPAdapterFaceIDXL(IPAdapterFaceID):
515
  do_classifier_free_guidance=True,
516
  negative_prompt=negative_prompt,
517
  )
 
 
 
518
 
519
  prompt_embeds = torch.cat([prompt_embeds, image_prompt_embeds], dim=1)
520
  negative_prompt_embeds = torch.cat([negative_prompt_embeds, uncond_image_prompt_embeds], dim=1)
 
515
  do_classifier_free_guidance=True,
516
  negative_prompt=negative_prompt,
517
  )
518
+
519
+ image_prompt_embeds = image_prompt_embeds.to(dtype=prompt_embeds.dtype)
520
+ uncond_image_prompt_embeds = uncond_image_prompt_embeds.to(dtype=negative_prompt_embeds.dtype)
521
 
522
  prompt_embeds = torch.cat([prompt_embeds, image_prompt_embeds], dim=1)
523
  negative_prompt_embeds = torch.cat([negative_prompt_embeds, uncond_image_prompt_embeds], dim=1)