[Network] Use NetworkAddress for dist_init_method and loopback fallbacks (#20657)

Co-authored-by: hnyls2002 <lsyincs@gmail.com>
This commit is contained in:
psaab
2026-03-16 19:59:49 -07:00
committed by GitHub
parent 71a54c1c42
commit 9f56b471aa
9 changed files with 79 additions and 44 deletions

View File

@@ -288,7 +288,8 @@ class DataParallelController:
# Determine the endpoint for inter-node communication
if server_args.dist_init_addr is None:
na = NetworkAddress(
"127.0.0.1", server_args.port + DP_ATTENTION_HANDSHAKE_PORT_DELTA
server_args.host or "127.0.0.1",
server_args.port + DP_ATTENTION_HANDSHAKE_PORT_DELTA,
)
else:
na = NetworkAddress.parse(server_args.dist_init_addr)