diff --git a/python/sglang/multimodal_gen/runtime/models/dits/causal_wanvideo.py b/python/sglang/multimodal_gen/runtime/models/dits/causal_wanvideo.py index 279aca9f2..1b5bdc28b 100644 --- a/python/sglang/multimodal_gen/runtime/models/dits/causal_wanvideo.py +++ b/python/sglang/multimodal_gen/runtime/models/dits/causal_wanvideo.py @@ -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(