fix mtp error

wangxiyuan · wangxiyuan · commit 307af29be83e · 2025-12-02T17:03:59.000+08:00
Signed-off-by: wangxiyuan &lt;wangxiyuan1007@gmail.com&gt;
diff --git a/vllm_ascend/worker/model_runner_v1.py b/vllm_ascend/worker/model_runner_v1.py
@@ -2510,7 +2510,7 @@ def sample_tokens(
                     valid_sampled_token_ids = sampled_token_ids.tolist()
                 else:
                     # Includes spec decode tokens. It's a numpy array
-                    valid_sampled_token_ids = self.rejection_sampler.parse_output(
+                    valid_sampled_token_ids, _ = self.rejection_sampler.parse_output(
                         sampled_token_ids,
                         self.input_batch.vocab_size,
                     )

Original file line number	Diff line number	Diff line change
`@@ -2510,7 +2510,7 @@ def sample_tokens(`
`2510`	`2510`	`valid_sampled_token_ids = sampled_token_ids.tolist()`
`2511`	`2511`	`else:`
`2512`	`2512`	`# Includes spec decode tokens. It's a numpy array`
`2513`		`- valid_sampled_token_ids = self.rejection_sampler.parse_output(`
	`2513`	`+ valid_sampled_token_ids, _ = self.rejection_sampler.parse_output(`
`2514`	`2514`	`sampled_token_ids,`
`2515`	`2515`	`self.input_batch.vocab_size,`
`2516`	`2516`	`)`