xinyun/vllm
1
0
Fork 0
mirror of https://git.datalinker.icu/vllm-project/vllm.git synced 2025-12-16 05:35:01 +08:00
Code Issues Packages Projects Releases Wiki Activity

[Core] init_distributed_environment align with init_process_group(#4014)

Browse Source 
[Core][Distributed] make init_distributed_environment compatible with init_process_group (#4014)
This commit is contained in:
youkaichao 2024-04-11 14:00:48 -07:00 committed by GitHub
parent a10d3056da
commit 559eb852f8
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
vllm/distributed/parallel_state.py
View File

@ -39,9 +39,9 @@ _PIPELINE_GLOBAL_RANKS = None
def init_distributed_environment( 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, local_rank: int = -1,
backend: str = "nccl", backend: str = "nccl",
): ):