Skip to content

Commit f8a2d55

Browse files
committed
Fixed a bug that results in white noise for generation
1 parent e4dfe2b commit f8a2d55

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

src/diffusers/models/transformers/transformer_wan_animate.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -340,8 +340,8 @@ def forward(self, face_image: torch.Tensor, channel_dim: int = 1) -> torch.Tenso
340340
weight = self.motion_synthesis_weight + 1e-8
341341
# Upcast the QR orthogonalization operation to FP32
342342
original_motion_dtype = motion_feat.dtype
343-
motion_feat = motion_feat.to(torch.float32)
344-
weight = weight.to(torch.float32)
343+
motion_feat = motion_feat.to(weight.dtype)
344+
# weight = weight.to(torch.float32)
345345

346346
Q = torch.linalg.qr(weight)[0].to(device=motion_feat.device)
347347

0 commit comments

Comments
 (0)