From cb489682943153b5b6103d436474a8e7a9e28ca2 Mon Sep 17 00:00:00 2001 From: Canlin Guo Date: Wed, 28 Jan 2026 18:42:39 +0800 Subject: [PATCH] [Bugfix] Modify output to model_runner_output (#1026) Signed-off-by: gcanlin --- vllm_omni/worker/npu/npu_ar_model_runner.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/vllm_omni/worker/npu/npu_ar_model_runner.py b/vllm_omni/worker/npu/npu_ar_model_runner.py index 0eedb61fb..1bb53b662 100644 --- a/vllm_omni/worker/npu/npu_ar_model_runner.py +++ b/vllm_omni/worker/npu/npu_ar_model_runner.py @@ -438,7 +438,7 @@ class NPUARModelRunner(OmniNPUModelRunner): pooler_output=(pooler_output if self.vllm_config.model_config.engine_output_type != "text" else None), kv_connector_output=kv_connector_output, ) - output.kv_extracted_req_ids = kv_extracted_req_ids + model_runner_output.kv_extracted_req_ids = kv_extracted_req_ids # -------------------------------------- Omni-new ------------------------------------------------- durations = ProfileExecuteDuration().pop_captured_sync()