Fix BailingMoeV3 reading enable_dp_lm_head off live topology instead of config (#36584)

2026-08-26 18:41:39 -07:00
parent a8b72c4c34
commit 15688fea7d
2 changed files with 2 additions and 2 deletions
@@ -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)