[Bugfix] Modify output to model_runner_output (#1026)

Signed-off-by: gcanlin <canlinguosdu@gmail.com>
This commit is contained in:
Canlin Guo authored and GitHub committed 2026-01-28 18:42:39 +08:00
1 parent 91a3d465ee
commit cb48968294
1 file changed
+1 -1
+1 -1
View File
@@ -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()