[Bugfix] Use uds for zmq address if not set --stage-id (#1522)

Signed-off-by: wuhang <wuhang6@huawei.com>

vllm_omni/entrypoints/omni.py

@@ -371,9 +371,13 @@ class OmniBase:
             # Allocate endpoints for each stage
             total_stages = len(self.stage_configs)
             self._handshake_endpoints = {}
+
+            # If --stage-id is not set, use local_only mode
+            local_only = self._single_stage_id is None
+
             for sid in range(total_stages):
-                in_endpoint = get_engine_client_zmq_addr(local_only=False, host=self._zmq_master_address)
-                out_endpoint = get_engine_client_zmq_addr(local_only=False, host=self._zmq_master_address)
+                in_endpoint = get_engine_client_zmq_addr(local_only=local_only, host=self._zmq_master_address)
+                out_endpoint = get_engine_client_zmq_addr(local_only=local_only, host=self._zmq_master_address)
                 self._handshake_endpoints[sid] = (in_endpoint, out_endpoint)
                 logger.debug(
                     f"[{self._name}] Allocated endpoints for stage-{sid}: in={in_endpoint}, out={out_endpoint}"