[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
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -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,
) )