revi13 commited on
Commit
406843b
·
verified ·
1 Parent(s): 00f8511

Update ip_adapter/ip_adapter_faceid.py

Browse files
Files changed (1) hide show
  1. ip_adapter/ip_adapter_faceid.py +3 -3
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,