From 38f58064eb3599b98db5fc2e322027f40ffae17a Mon Sep 17 00:00:00 2001 From: Baoyuan Qi Date: Thu, 27 Nov 2025 20:26:51 +0800 Subject: [PATCH] [fix] Remove redundant parameter passing (#90) MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit Signed-off-by: 齐保元 --- vllm_omni/entrypoints/omni_stage.py | 14 ++++---------- 1 file changed, 4 insertions(+), 10 deletions(-) diff --git a/vllm_omni/entrypoints/omni_stage.py b/vllm_omni/entrypoints/omni_stage.py index 481c48213..ad89b4cbb 100644 --- a/vllm_omni/entrypoints/omni_stage.py +++ b/vllm_omni/entrypoints/omni_stage.py @@ -138,9 +138,6 @@ class OmniStage: self, model, stage_payload, - self._in_q, - self._out_q, - self._log_file, batch_timeout, ), ) @@ -487,21 +484,15 @@ def _stage_worker_async_entry( omni_stage: OmniStage, model: str, stage_payload: dict[str, Any], - in_q: mp.Queue, - out_q: mp.Queue, - log_file: Optional[str] = None, batch_timeout: int = 10, ) -> None: - asyncio.run(_stage_worker_async(omni_stage, model, stage_payload, in_q, out_q, log_file, batch_timeout)) + asyncio.run(_stage_worker_async(omni_stage, model, stage_payload, batch_timeout)) async def _stage_worker_async( omni_stage: OmniStage, model: str, stage_payload: dict[str, Any], - in_q: mp.Queue, - out_q: mp.Queue, - log_file: Optional[str] = None, batch_timeout: int = 10, ) -> None: """Stage worker entry: device setup, LLM init, batching, SHM IPC.""" @@ -523,6 +514,9 @@ async def _stage_worker_async( runtime_cfg = stage_payload.get("runtime", {}) shm_threshold_bytes = int(stage_payload.get("shm_threshold_bytes", 65536)) + log_file = omni_stage._log_file + in_q = omni_stage._in_q + out_q = omni_stage._out_q # Per-stage file logger (optional) try: if log_file: