diff --git a/vllm/config/vllm.py b/vllm/config/vllm.py index 334b5890af..51b8b1019c 100644 --- a/vllm/config/vllm.py +++ b/vllm/config/vllm.py @@ -2464,10 +2464,6 @@ class VllmConfig: model_config = self.model_config speculative_config = self.speculative_config - if self.parallel_config.prefill_context_parallel_size > 1 and not ( - model_config is not None and model_config.use_mla - ): - unsupported.append("prefill context parallelism") if self.compilation_config.mode == CompilationMode.STOCK_TORCH_COMPILE: unsupported.append("stock torch.compile")