Lint

Author: Qirui Yang

vllm/v1/executor/multiproc_executor.py

@@ -63,10 +63,12 @@ def _init_executor(self) -> None:
         tensor_parallel_size = self.parallel_config.tensor_parallel_size
         context_parallel_size = self.parallel_config.context_parallel_size
         pp_parallel_size = self.parallel_config.pipeline_parallel_size
-        assert self.world_size == tensor_parallel_size * context_parallel_size * pp_parallel_size, (
-            f"world_size ({self.world_size}) must be equal to the "
-            f"tensor_parallel_size ({tensor_parallel_size}) x pipeline"
-            f"_parallel_size ({pp_parallel_size}). ")
+        assert self.world_size == (
+            tensor_parallel_size * context_parallel_size * pp_parallel_size), (
+                f"world_size ({self.world_size}) must be equal to the "
+                f"tensor_parallel_size ({tensor_parallel_size}) x "
+                f"context_parallel_size ({context_parallel_size}) x "
+                f"pipeline_parallel_size ({pp_parallel_size}). ")
 
         # Set multiprocessing envs
         set_multiprocessing_worker_envs()