diff --git a/python/sglang/srt/arg_groups/speculative_hook.py b/python/sglang/srt/arg_groups/speculative_hook.py index 5f8d5c8c9..390c1f94f 100644 --- a/python/sglang/srt/arg_groups/speculative_hook.py +++ b/python/sglang/srt/arg_groups/speculative_hook.py @@ -559,10 +559,10 @@ def _handle_dspark(server_args: ServerArgs) -> None: if cfg.enable_dp_attention and cfg.dp_size > 1: if not cfg.enable_dp_lm_head: raise ValueError("DSpark with dp attention requires --enable-dp-lm-head.") - if not _is_npu and cfg.moe_a2a_backend not in ("none", "megamoe"): + if not _is_npu and cfg.moe_a2a_backend not in ("none", "megamoe", "mori"): raise ValueError( "DSpark with dp attention supports moe_a2a_backend 'none' " - "(built-in TP MoE) or 'megamoe', got " + "(built-in TP MoE), 'megamoe', or 'mori', got " f"{cfg.moe_a2a_backend!r}." ) if not _is_npu and cfg.moe_a2a_backend != "none":