[Bugfix] Fix ipv6 address parsing bug (#3641)

2024-03-27 02:58:20 +08:00
parent 8af890a865
commit a979d9771e
1 changed files with 1 additions and 1 deletions
--- a/vllm/utils.py
+++ b/vllm/utils.py
@ -205,7 +205,7 @@ def get_ip() -> str:


 def get_distributed_init_method(ip: str, port: int) -> str:
-    return f"tcp://{ip}:{port}"
+    return f"tcp://[{ip}]:{port}"


 def get_open_port() -> int: