[ci][test] fix pp test failure (vllm-project#7945)

Signed-off-by: Alvant <alvasian@yandex.ru>
compressa-ai · Oct 26, 2024 · c5077c2 · c5077c2
1 parent 4b6fa2b
commit c5077c2
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/vllm/executor/multiproc_gpu_executor.py b/vllm/executor/multiproc_gpu_executor.py
@@ -126,7 +126,7 @@ def shutdown(signum, frame):
                           max_parallel_loading_workers)
 
     def _check_executor_parameters(self):
-        world_size = self.parallel_config.tensor_parallel_size
+        world_size = self.parallel_config.world_size
         tensor_parallel_size = self.parallel_config.tensor_parallel_size
 
         # Set CUDA_VISIBLE_DEVICES for the driver, inherited by workers