[diffusion] Fix torch.zeros typo in causal wan (#21250)

2026-03-24 14:39:16 +08:00
parent d9f97b2115
commit 69f02e36e8
1 changed files with 1 additions and 1 deletions
@@ -397,7 +397,7 @@ class CausalWanTransformerBlock(nn.Module):
        attn_output, _ = self.to_out(attn_output)
        attn_output = attn_output.squeeze(1)

-        null_shift = null_scale = torch.zeroes(
+        null_shift = null_scale = torch.zeros(
            (1,), device=hidden_states.device, dtype=hidden_states.dtype
        )
        norm_hidden_states, hidden_states = self.self_attn_residual_norm(