Unverified Commit f0359fff authored by Jiangyun Zhu's avatar Jiangyun Zhu Committed by GitHub
Browse files

[Bugfix] fix qwen3-next crash (#28202)


Signed-off-by: default avatarzjy0516 <riverclouds.zhu@qq.com>
parent 798c7beb
...@@ -587,7 +587,7 @@ class Qwen3NextGatedDeltaNet(nn.Module, MambaBase): ...@@ -587,7 +587,7 @@ class Qwen3NextGatedDeltaNet(nn.Module, MambaBase):
self.conv1d.bias, self.conv1d.bias,
self.activation, self.activation,
conv_state_indices=non_spec_state_indices_tensor[ conv_state_indices=non_spec_state_indices_tensor[
: attn_metadata.num_decodes : attn_metadata.num_actual_tokens
], ],
validate_data=True, validate_data=True,
) )
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment