[Core] init_distributed_environment align with init_process_group(#4014)

[Core][Distributed] make init_distributed_environment compatible with init_process_group (#4014)

[Core] init_distributed_environment align with init_process_group(#4014)
[Core][Distributed] make init_distributed_environment compatible with init_process_group (#4014)
559eb852 · youkaichao · GitHub · a10d3056 · 559eb852
Unverified Commit 559eb852 authored Apr 11, 2024 by youkaichao Committed by GitHub Apr 11, 2024
Show whitespace changes
Inline Side-by-side

Showing with 3 additions and 3 deletions

vllm/distributed/parallel_state.py vllm/distributed/parallel_state.py +3 -3

No files found.
--- a/vllm/distributed/parallel_state.py
+++ b/vllm/distributed/parallel_state.py
@@ -39,9 +39,9 @@ _PIPELINE_GLOBAL_RANKS = None
 def init_distributed_environment(
-    world_size: int,
+    world_size: int = -1,
-    rank: int,
+    rank: int = -1,
-    distributed_init_method: Optional[str] = None,
+    distributed_init_method: str = "env://",
    local_rank: int = -1,
    backend: str = "nccl",
 ):