[Feature] Speculative decoding support lookahead (#9873)

Co-authored-by: a4zhangfei <a4zhangfei@qq.com>
Co-authored-by: Qiaolin-Yu <liin1211@outlook.com>
This commit is contained in:
Zhihao Zhang
2025-09-18 16:42:41 -07:00
committed by GitHub
co-authored by a4zhangfei Qiaolin-Yu
parent 2a2ff9a840
commit e7bc600304
30 changed files with 2058 additions and 32 deletions
+2 -1
View File
@@ -1110,7 +1110,8 @@ def sample_sharegpt_requests(
add_generation_prompt=True,
tokenize=False,
)
prompt = prompt.replace(tokenizer.bos_token, "")
if tokenizer.bos_token:
prompt = prompt.replace(tokenizer.bos_token, "")
prompt_token_ids = tokenizer.encode(prompt)
completion = dataset[i][1]