Spaces:
Running
on
Zero
Running
on
Zero
Commit
·
beb29f7
1
Parent(s):
7f97ed5
Fix wrong negative_prompt_embeds_
Browse files
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 +
|
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])
|