From 787dfed0e9298457c610c000bcef4645623ea125 Mon Sep 17 00:00:00 2001 From: Jintao Huang Date: Mon, 4 May 2026 18:29:00 +0800 Subject: [PATCH] update --- src/mcore_bridge/model/gpts/qwen3_next.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/mcore_bridge/model/gpts/qwen3_next.py b/src/mcore_bridge/model/gpts/qwen3_next.py index 08cce14..5380434 100644 --- a/src/mcore_bridge/model/gpts/qwen3_next.py +++ b/src/mcore_bridge/model/gpts/qwen3_next.py @@ -497,7 +497,7 @@ def forward(self, hidden_states: torch.Tensor, **kwargs): hidden_states = new_hidden_states else: hidden_states = hidden_states.transpose(0, 1) - attention_mask = resolve_hf_attention_mask(kwargs) + attention_mask = resolve_gdn_attention_mask(kwargs) res = super().forward(hidden_states=hidden_states, attention_mask=attention_mask) if thd_format: res = res[attention_mask][:, None]