[Core] init_distributed_environment align with init_process_group(vll…

…m-project#4014) [Core][Distributed] make init_distributed_environment compatible with init_process_group (vllm-project#4014)
neuralmagic · Apr 12, 2024 · 9f93716 · 9f93716
1 parent 1d859cc
commit 9f93716
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/vllm/distributed/parallel_state.py b/vllm/distributed/parallel_state.py
@@ -39,9 +39,9 @@
 
 
 def init_distributed_environment(
-    world_size: int,
-    rank: int,
-    distributed_init_method: Optional[str] = None,
+    world_size: int = -1,
+    rank: int = -1,
+    distributed_init_method: str = "env://",
     local_rank: int = -1,
     backend: str = "nccl",
 ):