Skip to content

Commit f8cd5b1

Browse files
authored
Fix bf15/fp16 for pipeline_wan_vace.py
1 parent a6d2fc2 commit f8cd5b1

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/diffusers/pipelines/wan/pipeline_wan_vace.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -526,7 +526,7 @@ def prepare_video_latents(
526526
latents = ((latents.float() - latents_mean) * latents_std).to(vae_dtype)
527527
else:
528528
mask = mask.to(dtype=vae_dtype)
529-
mask = torch.where(mask > 0.5, 1.0, 0.0)
529+
mask = torch.where(mask > 0.5, 1.0, 0.0).to(dtype=vae_dtype)
530530
inactive = video * (1 - mask)
531531
reactive = video * mask
532532
inactive = retrieve_latents(self.vae.encode(inactive), generator, sample_mode="argmax")

0 commit comments

Comments
 (0)