We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent a6d2fc2 commit f8cd5b1Copy full SHA for f8cd5b1
src/diffusers/pipelines/wan/pipeline_wan_vace.py
@@ -526,7 +526,7 @@ def prepare_video_latents(
526
latents = ((latents.float() - latents_mean) * latents_std).to(vae_dtype)
527
else:
528
mask = mask.to(dtype=vae_dtype)
529
- mask = torch.where(mask > 0.5, 1.0, 0.0)
+ mask = torch.where(mask > 0.5, 1.0, 0.0).to(dtype=vae_dtype)
530
inactive = video * (1 - mask)
531
reactive = video * mask
532
inactive = retrieve_latents(self.vae.encode(inactive), generator, sample_mode="argmax")
0 commit comments