diff --git a/vllm/config.py b/vllm/config.py index 3e6aa2a93e6a7..d649eb75033f9 100644 --- a/vllm/config.py +++ b/vllm/config.py @@ -2108,6 +2108,15 @@ class ParallelConfig: 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(