[ci][test] fix pp test failure (#7945)

2024-08-28 01:27:07 -07:00
parent e3580537a4
commit f52a43a8b9
1 changed files with 1 additions and 1 deletions
--- a/vllm/executor/multiproc_gpu_executor.py
+++ b/vllm/executor/multiproc_gpu_executor.py
@ -126,7 +126,7 @@ class MultiprocessingGPUExecutor(DistributedGPUExecutor):
                          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