diff --git a/python/sglang/srt/speculative/frozen_kv_mtp_utils.py b/python/sglang/srt/speculative/frozen_kv_mtp_utils.py index 71ebc4c99..043d8b63f 100644 --- a/python/sglang/srt/speculative/frozen_kv_mtp_utils.py +++ b/python/sglang/srt/speculative/frozen_kv_mtp_utils.py @@ -140,7 +140,7 @@ def select_last_verified_seed( counts = draft_input.num_accept_tokens.to(torch.long) last_indices = torch.cumsum(counts, dim=0) - 1 return ( - draft_input.bonus_tokens[last_indices], + draft_input.input_ids[last_indices], draft_input.hidden_states[last_indices], )