Skip to content

Commit 463d4b7

Browse files
authored
Fix DeepEP cannot run on latest master (#5567)
1 parent 9924bbe commit 463d4b7

File tree

1 file changed

+2
-0
lines changed
  • python/sglang/srt/layers/moe/ep_moe

1 file changed

+2
-0
lines changed

python/sglang/srt/layers/moe/ep_moe/layer.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -802,6 +802,7 @@ def __init__(
802802
correction_bias: Optional[torch.Tensor] = None,
803803
custom_routing_function: Optional[Callable] = None,
804804
activation: str = "silu",
805+
routed_scaling_factor: Optional[float] = None,
805806
deepep_mode: DeepEPMode = DeepEPMode.auto,
806807
):
807808
super().__init__(
@@ -820,6 +821,7 @@ def __init__(
820821
correction_bias,
821822
custom_routing_function,
822823
activation,
824+
routed_scaling_factor,
823825
)
824826
self.deepep_mode = deepep_mode
825827
if self.deepep_mode.enable_low_latency():

0 commit comments

Comments
 (0)