From 6beb4434e18036a3052603edb77320739cbbe953 Mon Sep 17 00:00:00 2001 From: Mengqing Cao Date: Mon, 15 Dec 2025 11:10:39 +0800 Subject: [PATCH] [CI][Bugfix] Fix scheduleroutput has no attr get error in prompt logprobs (#4998) ### What this PR does / why we need it? Fix scheduleroutput has no attr get error in prompt logprobs Fix https://github.com/vllm-project/vllm-ascend/actions/runs/20194753373/job/57977131870 ### How was this patch tested? CI passed with existing test. - vLLM version: v0.12.0 - vLLM main: https://github.com/vllm-project/vllm/commit/ad32e3e19ccf0526cb6744a5fed09a138a5fb2f9 Signed-off-by: MengqingCao --- vllm_ascend/worker/model_runner_v1.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/vllm_ascend/worker/model_runner_v1.py b/vllm_ascend/worker/model_runner_v1.py index 0b514e60..c56a4562 100644 --- a/vllm_ascend/worker/model_runner_v1.py +++ b/vllm_ascend/worker/model_runner_v1.py @@ -1762,7 +1762,7 @@ class NPUModelRunner(GPUModelRunner): # Compute prompt logprobs if needed. prompt_logprobs_dict = self._get_prompt_logprobs_dict( hidden_states[:scheduler_output.total_num_scheduled_tokens], - scheduler_output, + scheduler_output.num_scheduled_tokens, ) num_sampled_tokens = sampler_output.sampled_token_ids.shape[0]