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

Signed-off-by: zjy0516 <riverclouds.zhu@qq.com>
This commit is contained in:
Jiangyun Zhu
2025-11-11 16:24:28 +08:00
committed by GitHub
parent 798c7bebca
commit f0359fffa4

View File

@@ -587,7 +587,7 @@ class Qwen3NextGatedDeltaNet(nn.Module, MambaBase):
self.conv1d.bias,
self.activation,
conv_state_indices=non_spec_state_indices_tensor[
: attn_metadata.num_decodes
: attn_metadata.num_actual_tokens
],
validate_data=True,
)