From 4ea041fbbd519bb6086b5bdef1f5d1618e806621 Mon Sep 17 00:00:00 2001
From: Joe Runde <Joseph.Runde@ibm.com>
Date: Tue, 26 Mar 2024 15:10:13 -0600
Subject: [PATCH] Squash 3645

---
 vllm/utils.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/vllm/utils.py b/vllm/utils.py
index 653f1af3f..bffb7a2a3 100644
--- 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: