adaface-neurips commited on
Commit
beb29f7
·
1 Parent(s): 7f97ed5

Fix wrong negative_prompt_embeds_

Browse files
Files changed (1) hide show
  1. faceadapter/face_adapter.py +1 -2
faceadapter/face_adapter.py CHANGED
@@ -312,13 +312,12 @@ class FaceAdapterPlusForVideoLora(FaceAdapterLora):
312
  )
313
 
314
  if adaface_prompt_embeds is not None and adaface_prompt_embeds[0] is not None:
315
- negative_prompt_embeds0 = negative_prompt_embeds_
316
  adaface_prompt_embeds, negative_prompt_embeds_ = adaface_prompt_embeds
317
  # self.torch_type == torch.float16. adaface_prompt_embeds is torch.float32.
318
  prompt_embeds_ = adaface_prompt_embeds.repeat(num_samples, 1, 1).to(dtype=self.torch_type)
319
  negative_prompt_embeds_ = negative_prompt_embeds_.repeat(num_samples, 1, 1).to(dtype=self.torch_type)
320
  if adaface_power_scale != 1.0:
321
- prompt_embeds_ = prompt_embeds_ * adaface_power_scale + negative_prompt_embeds0 * (1 - adaface_power_scale)
322
 
323
  # Note to balance image_prompt_embeds with uncond_image_prompt_embeds after scaling.
324
  image_prompt_embeds_begin = image_prompt_embeds * image_embed_cfg_scales[0] + uncond_image_prompt_embeds * (1 - image_embed_cfg_scales[0])
 
312
  )
313
 
314
  if adaface_prompt_embeds is not None and adaface_prompt_embeds[0] is not None:
 
315
  adaface_prompt_embeds, negative_prompt_embeds_ = adaface_prompt_embeds
316
  # self.torch_type == torch.float16. adaface_prompt_embeds is torch.float32.
317
  prompt_embeds_ = adaface_prompt_embeds.repeat(num_samples, 1, 1).to(dtype=self.torch_type)
318
  negative_prompt_embeds_ = negative_prompt_embeds_.repeat(num_samples, 1, 1).to(dtype=self.torch_type)
319
  if adaface_power_scale != 1.0:
320
+ prompt_embeds_ = prompt_embeds_ * adaface_power_scale + negative_prompt_embeds_ * (1 - adaface_power_scale)
321
 
322
  # Note to balance image_prompt_embeds with uncond_image_prompt_embeds after scaling.
323
  image_prompt_embeds_begin = image_prompt_embeds * image_embed_cfg_scales[0] + uncond_image_prompt_embeds * (1 - image_embed_cfg_scales[0])