Squash 3645

IBM · Mar 26, 2024 · 4ea041f · 4ea041f
1 parent abe950a
commit 4ea041f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/vllm/utils.py b/vllm/utils.py
@@ -206,7 +206,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}" if ":" in ip else f"tcp://{ip}:{port}"
 
 
 def get_open_port() -> int: