From 33586d845ce62104f68dae8da34a38c2b557618b Mon Sep 17 00:00:00 2001 From: Alex Brooks Date: Sun, 3 May 2026 23:50:31 -0600 Subject: [PATCH] [Bugfix] Use get_open_ports_list for stage ports in OmniMasterServer (#3333) Signed-off-by: Alex Brooks --- vllm_omni/engine/stage_engine_startup.py | 6 ++---- 1 file changed, 2 insertions(+), 4 deletions(-) diff --git a/vllm_omni/engine/stage_engine_startup.py b/vllm_omni/engine/stage_engine_startup.py index 45e6638c4..568fcba20 100644 --- a/vllm_omni/engine/stage_engine_startup.py +++ b/vllm_omni/engine/stage_engine_startup.py @@ -14,7 +14,7 @@ import zmq from omegaconf import OmegaConf from vllm.config import CacheConfig, VllmConfig from vllm.logger import init_logger -from vllm.utils.network_utils import get_open_port, zmq_socket_ctx +from vllm.utils.network_utils import get_open_ports_list, zmq_socket_ctx from vllm.v1.engine.coordinator import DPCoordinator from vllm.v1.engine.utils import ( STARTUP_POLL_PERIOD_MS, @@ -120,9 +120,7 @@ class OmniMasterServer: for sid in stage_ids: self._stage_config_events[sid] = threading.Event() self._stage_coordinator_addresses[sid] = StageCoordinatorAddresses() - hs_port = get_open_port() - inp_port = get_open_port() - out_port = get_open_port() + hs_port, inp_port, out_port = get_open_ports_list(count=3) self._allocations[sid] = StageAllocation( handshake_bind_address=f"tcp://{master_address}:{hs_port}", handshake_connect_address=f"tcp://{master_address}:{hs_port}",