Skip to content

Commit c01d2bf

Browse files
chengzeyiyiyixuxu
andauthored
Update src/diffusers/pipelines/wan/pipeline_wan_i2v.py
Co-authored-by: YiYi Xu <[email protected]>
1 parent 482415c commit c01d2bf

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/diffusers/pipelines/wan/pipeline_wan_i2v.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -113,7 +113,7 @@ def retrieve_latents(
113113
latents_mean: torch.Tensor,
114114
latents_std: torch.Tensor,
115115
generator: Optional[torch.Generator] = None,
116-
sample_mode: str = "none",
116+
sample_mode: str = "argmax",
117117
):
118118
if hasattr(encoder_output, "latent_dist") and sample_mode == "none":
119119
return (encoder_output.latent_dist.mean - latents_mean) * latents_std

0 commit comments

Comments
 (0)