[hotfix] fix merge conflicts in FlashInferEPMoE (#8405)

2025-07-27 01:24:10 -07:00
parent 44d600cd67
commit 4d921f2b79
1 changed files with 1 additions and 0 deletions
--- a/python/sglang/srt/layers/moe/ep_moe/layer.py
+++ b/python/sglang/srt/layers/moe/ep_moe/layer.py
@@ -1236,6 +1236,7 @@ class FlashInferEPMoE(EPMoE):
        self.num_expert_group = num_expert_group
        self.topk_group = topk_group
        self.correction_bias = correction_bias
+        self.use_flashinfer_trtllm_moe = use_flashinfer_trtllm_moe

    def forward(self, hidden_states: torch.Tensor, router_logits: torch.Tensor):
        assert use_flashinfer_trtllm_moe