Improve code styles (#4021)

2025-03-03 03:20:23 -08:00
parent ac2387279e
commit 66301e124f
14 changed files with 88 additions and 243 deletions
--- a/python/sglang/srt/layers/moe/ep_moe/layer.py
+++ b/python/sglang/srt/layers/moe/ep_moe/layer.py
@@ -181,7 +181,6 @@ class EPMoE(torch.nn.Module):

    def forward(self, hidden_states: torch.Tensor, router_logits: torch.Tensor):
        assert self.quant_method is not None
-        assert self.activation == "silu"

        if self.grouped_gemm_runner is None:
            self.grouped_gemm_runner = GroupedGemmRunner(