From 161743e3191c5312b11656ea7f7a1eeca2d9fac2 Mon Sep 17 00:00:00 2001 From: Jintao Huang Date: Tue, 5 May 2026 19:00:55 +0800 Subject: [PATCH] fix kwargs --- 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 9ce59bc..9b135cd 100644 --- a/src/mcore_bridge/model/gpts/qwen3_next.py +++ b/src/mcore_bridge/model/gpts/qwen3_next.py @@ -301,7 +301,7 @@ def nvtx_range_push(*args, **kwargs): k_pos_emb, config=self.config, cu_seqlens=cu_seqlens_kv, - **kwargs, + cp_group=self.pg_collection.cp, ) # TODO, can apply positional embedding to value_layer so it has