mirror of
https://github.com/vllm-project/vllm.git
synced 2026-08-10 07:48:09 +00:00
[Perf] Skip detokenization in online beam search (#46422)
Signed-off-by: Guy Stone <[email protected]> Signed-off-by: Guy Stone <[email protected]> Co-authored-by: Claude <[email protected]>
This commit is contained in:
@@ -61,6 +61,7 @@ class BeamSearchOnlineMixin(ABC):
|
||||
logprobs=logprobs_num,
|
||||
max_tokens=1,
|
||||
temperature=temperature,
|
||||
detokenize=False,
|
||||
)
|
||||
all_beams = [
|
||||
BeamSearchSequence(
|
||||
|
||||
Reference in New Issue
Block a user