"vllm/vscode:/vscode.git/clone" did not exist on "4fb8beefaa8b2c4bd2cd3b336b01ff006dc98bdc"
Unverified Commit 2ec88272 authored by Sky Lee's avatar Sky Lee Committed by GitHub
Browse files

[Bugfix] Qwen-vl output is inconsistent in speculative decoding (#10350)

parent b40cf640
......@@ -353,6 +353,7 @@ class BatchExpansionTop1Scorer(SpeculativeScorer):
seq_data = seq_group_metadata.seq_data[seq_id]
prompt_token_ids = seq_data.prompt_token_ids_array
new_output_token_ids = [*seq_data.get_output_token_ids(), *token_ids]
mrope_position_delta = seq_data.mrope_position_delta
new_seq_data_dict = {
target_seq_id:
......@@ -368,6 +369,7 @@ class BatchExpansionTop1Scorer(SpeculativeScorer):
# the kv cache is filled by a previous batch in the batch expansion.
for data in new_seq_data_dict.values():
data.update_num_computed_tokens(data.get_len() - 1)
data.mrope_position_delta = mrope_position_delta
return SequenceGroupMetadata(
request_id=seq_group_metadata.request_id,
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment