From 5aa6efb9a5929d9b4b6b2c231b81f63587e83c77 Mon Sep 17 00:00:00 2001 From: Lifu Huang Date: Thu, 24 Apr 2025 22:49:30 -0700 Subject: [PATCH] [Misc] Clean up redundant code in uniproc_executor.py (#16762) Signed-off-by: Lifu Huang --- vllm/executor/uniproc_executor.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/vllm/executor/uniproc_executor.py b/vllm/executor/uniproc_executor.py index 8c004c790fcb..2e4b47c1e24a 100644 --- a/vllm/executor/uniproc_executor.py +++ b/vllm/executor/uniproc_executor.py @@ -34,13 +34,13 @@ class UniProcExecutor(ExecutorBase): if len(device_info) > 1: local_rank = int(device_info[1]) rank = 0 + is_driver_worker = True kwargs = dict( vllm_config=self.vllm_config, local_rank=local_rank, rank=rank, distributed_init_method=distributed_init_method, - is_driver_worker=(not self.parallel_config) - or (rank % self.parallel_config.tensor_parallel_size == 0), + is_driver_worker=is_driver_worker, ) self.collective_rpc("init_worker", args=([kwargs], )) self.collective_rpc("init_device")