From 95410fd8646f49fc6c1c916837e98d50af027a35 Mon Sep 17 00:00:00 2001
From: Nick Hill <nickhill@us.ibm.com>
Date: Tue, 26 Mar 2024 13:56:43 -0700
Subject: [PATCH] [BugFix] Fix ipv4 address parsing bug

#3641 was a fix for ipv6 but broke ipv4 case.
---
 vllm/utils.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/vllm/utils.py b/vllm/utils.py
index 4899b9074060..410883cdfaa2 100644
--- 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}" if ":" in ip else f"tcp://{ip}:{port}"
 
 
 def get_open_port() -> int: