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: