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

Hyper-Accel · Aug 28, 2024 · f52a43a · f52a43a
1 parent e358053
commit f52a43a
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