From f7dc281cb7d4e03d4a057a6c2aafb33540d5775a Mon Sep 17 00:00:00 2001 From: "rongfu.leng" Date: Thu, 17 Jul 2025 00:49:48 -0700 Subject: [PATCH] [misc][eplb] add valida ep or tp or dp Signed-off-by: rongfu.leng --- vllm/config.py | 9 +++++++++ 1 file changed, 9 insertions(+) diff --git a/vllm/config.py b/vllm/config.py index 22f740171369..dc4a219e2f0e 100644 --- a/vllm/config.py +++ b/vllm/config.py @@ -2063,6 +2063,15 @@ def __post_init__(self) -> None: raise ValueError( "num_redundant_experts must be non-negative, but got " f"{self.num_redundant_experts}.") + if not self.enable_expert_parallel: + raise ValueError( + "enable_expert_parallel must be True to use EPLB.") + if self.tensor_parallel_size * self.data_parallel_size <= 1: + raise ValueError( + "EPLB requires tensor_parallel_size or data_parallel_size " + f"to be greater than 1, but got " + f"TP={self.tensor_parallel_size},DP={self.data_parallel_size}." + ) else: if self.num_redundant_experts != 0: raise ValueError(