minke.yu/sglang
1
0
Fork 0
Code Issues Pull Requests Actions 12 Packages Projects Releases Wiki Activity

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

Browse Source
This commit is contained in:
Xiaoyu Zhang
2026-03-24 14:39:16 +08:00
committed by GitHub
parent d9f97b2115
commit 69f02e36e8
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
python/sglang/multimodal_gen/runtime/models/dits/causal_wanvideo.py
+1 -1
View File
@@ -397,7 +397,7 @@ class CausalWanTransformerBlock(nn.Module):
attn_output, _ = self.to_out(attn_output) attn_output, _ = self.to_out(attn_output)
attn_output = attn_output.squeeze(1) 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 (1,), device=hidden_states.device, dtype=hidden_states.dtype
) )
norm_hidden_states, hidden_states = self.self_attn_residual_norm( norm_hidden_states, hidden_states = self.self_attn_residual_norm(