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