File tree Expand file tree Collapse file tree 1 file changed +3
-3
lines changed Expand file tree Collapse file tree 1 file changed +3
-3
lines changed Original file line number Diff line number Diff line change 4141 DeviceCommunicatorBase )
4242from vllm .distributed .utils import StatelessProcessGroup
4343from vllm .logger import init_logger
44- from vllm .utils import (direct_register_custom_op , resolve_obj_by_qualname ,
45- supports_custom_op )
44+ from vllm .utils import (direct_register_custom_op , get_distributed_init_method ,
45+ resolve_obj_by_qualname , supports_custom_op )
4646
4747
4848@dataclass
@@ -929,7 +929,7 @@ def init_distributed_environment(
929929 world_size = parallel_config .world_size_across_dp
930930 ip = parallel_config .data_parallel_master_ip
931931 port = parallel_config .get_next_dp_init_port ()
932- distributed_init_method = f"tcp:// { ip } : { port } " # noqa
932+ distributed_init_method = get_distributed_init_method ( ip , port )
933933 logger .info (
934934 "Adjusting world_size=%d rank=%d distributed_init_method=%s for DP" ,
935935 world_size , rank , distributed_init_method )
You can’t perform that action at this time.
0 commit comments