Skip to content

Commit d644ebb

Browse files
lifuhuangdbyoung18
authored andcommitted
[Misc] Clean up redundant code in uniproc_executor.py (vllm-project#16762)
Signed-off-by: Lifu Huang <[email protected]>
1 parent f8bab54 commit d644ebb

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

vllm/executor/uniproc_executor.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -34,13 +34,13 @@ def _init_executor(self) -> None:
3434
if len(device_info) > 1:
3535
local_rank = int(device_info[1])
3636
rank = 0
37+
is_driver_worker = True
3738
kwargs = dict(
3839
vllm_config=self.vllm_config,
3940
local_rank=local_rank,
4041
rank=rank,
4142
distributed_init_method=distributed_init_method,
42-
is_driver_worker=(not self.parallel_config)
43-
or (rank % self.parallel_config.tensor_parallel_size == 0),
43+
is_driver_worker=is_driver_worker,
4444
)
4545
self.collective_rpc("init_worker", args=([kwargs], ))
4646
self.collective_rpc("init_device")

0 commit comments

Comments
 (0)