Spaces:
Sleeping
Sleeping
Update ip_adapter/ip_adapter_faceid.py
Browse files
ip_adapter/ip_adapter_faceid.py
CHANGED
|
@@ -404,8 +404,7 @@ class IPAdapterFaceIDPlus:
|
|
| 404 |
).images
|
| 405 |
|
| 406 |
return images
|
| 407 |
-
|
| 408 |
-
|
| 409 |
class IPAdapterFaceIDXL(IPAdapterFaceID):
|
| 410 |
"""SDXL"""
|
| 411 |
|
|
@@ -450,6 +449,7 @@ class IPAdapterFaceIDXL(IPAdapterFaceID):
|
|
| 450 |
negative_pooled_prompt_embeds,
|
| 451 |
) = self.pipe.encode_prompt(
|
| 452 |
prompt,
|
|
|
|
| 453 |
num_images_per_prompt=num_samples,
|
| 454 |
do_classifier_free_guidance=True,
|
| 455 |
negative_prompt=negative_prompt,
|
|
@@ -471,7 +471,6 @@ class IPAdapterFaceIDXL(IPAdapterFaceID):
|
|
| 471 |
|
| 472 |
return images
|
| 473 |
|
| 474 |
-
|
| 475 |
class IPAdapterFaceIDPlusXL(IPAdapterFaceIDPlus):
|
| 476 |
"""SDXL"""
|
| 477 |
|
|
@@ -520,6 +519,7 @@ class IPAdapterFaceIDPlusXL(IPAdapterFaceIDPlus):
|
|
| 520 |
negative_pooled_prompt_embeds,
|
| 521 |
) = self.pipe.encode_prompt(
|
| 522 |
prompt,
|
|
|
|
| 523 |
num_images_per_prompt=num_samples,
|
| 524 |
do_classifier_free_guidance=True,
|
| 525 |
negative_prompt=negative_prompt,
|
|
|
|
| 404 |
).images
|
| 405 |
|
| 406 |
return images
|
| 407 |
+
|
|
|
|
| 408 |
class IPAdapterFaceIDXL(IPAdapterFaceID):
|
| 409 |
"""SDXL"""
|
| 410 |
|
|
|
|
| 449 |
negative_pooled_prompt_embeds,
|
| 450 |
) = self.pipe.encode_prompt(
|
| 451 |
prompt,
|
| 452 |
+
device=self.device, # ← ここを追加
|
| 453 |
num_images_per_prompt=num_samples,
|
| 454 |
do_classifier_free_guidance=True,
|
| 455 |
negative_prompt=negative_prompt,
|
|
|
|
| 471 |
|
| 472 |
return images
|
| 473 |
|
|
|
|
| 474 |
class IPAdapterFaceIDPlusXL(IPAdapterFaceIDPlus):
|
| 475 |
"""SDXL"""
|
| 476 |
|
|
|
|
| 519 |
negative_pooled_prompt_embeds,
|
| 520 |
) = self.pipe.encode_prompt(
|
| 521 |
prompt,
|
| 522 |
+
device=self.device, # ← ここを追加
|
| 523 |
num_images_per_prompt=num_samples,
|
| 524 |
do_classifier_free_guidance=True,
|
| 525 |
negative_prompt=negative_prompt,
|