diff --git a/python/sglang/srt/layers/moe/ep_moe/layer.py b/python/sglang/srt/layers/moe/ep_moe/layer.py index 6f896e297..012a78f17 100644 --- a/python/sglang/srt/layers/moe/ep_moe/layer.py +++ b/python/sglang/srt/layers/moe/ep_moe/layer.py @@ -304,7 +304,7 @@ class EPMoE(FusedMoE): m_max * self.start_expert_id, BLOCK_SIZE=512, ) - return output + return output * self.routed_scaling_factor class DeepEPMoE(EPMoE):