From 15688fea7d8b45dbbfd26cdf5c970b31fd31abdc Mon Sep 17 00:00:00 2001 From: Alison Shao <54658187+alisonshao@users.noreply.github.com> Date: Wed, 26 Aug 2026 18:41:39 -0700 Subject: [PATCH] Fix BailingMoeV3 reading enable_dp_lm_head off live topology instead of config (#36584) --- python/sglang/srt/models/bailing_moe_v3.py | 2 +- test/registered/unit/models/test_shared_experts_fusion_gates.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/python/sglang/srt/models/bailing_moe_v3.py b/python/sglang/srt/models/bailing_moe_v3.py index 14317b338..bf660d593 100644 --- a/python/sglang/srt/models/bailing_moe_v3.py +++ b/python/sglang/srt/models/bailing_moe_v3.py @@ -1333,7 +1333,7 @@ class BailingMoeV3ForCausalLM(nn.Module): # in the logits processor. Accuracy-neutral on ling-v3. params_dtype=torch.bfloat16, quant_config=quant_config, - use_attn_tp_group=get_parallel().enable_dp_lm_head, + use_attn_tp_group=get_parallel().config.enable_dp_lm_head, ) ) self.logits_processor = LogitsProcessor(config) diff --git a/test/registered/unit/models/test_shared_experts_fusion_gates.py b/test/registered/unit/models/test_shared_experts_fusion_gates.py index d53b365e9..f698e2421 100644 --- a/test/registered/unit/models/test_shared_experts_fusion_gates.py +++ b/test/registered/unit/models/test_shared_experts_fusion_gates.py @@ -335,7 +335,7 @@ class TestBailingMoeV3Gate(_FusionGateCase): parallel = SimpleNamespace( tp_size=1, moe_ep_size=1, - enable_dp_lm_head=False, + config=SimpleNamespace(enable_dp_lm_head=False), ) with ( unittest.mock.patch.object(