xinyun/vllm
1
0
Fork 0
mirror of https://git.datalinker.icu/vllm-project/vllm.git synced 2025-12-13 23:55:44 +08:00
Code Issues Packages Projects Releases Wiki Activity

[Doc] Move todo out of beam search docstring (#17183)

Browse Source 
Signed-off-by: Alex-Brooks <Alex.Brooks@ibm.com>
This commit is contained in:
Alex Brooks 2025-04-25 05:44:58 -06:00 committed by GitHub
parent f851b84266
commit 7feae92c1f
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 2 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
vllm/entrypoints/llm.py
View File

@ -519,11 +519,9 @@ class LLM:
prompts: A list of prompts. Each prompt can be a string or a list prompts: A list of prompts. Each prompt can be a string or a list
of token IDs. of token IDs.
params: The beam search parameters. params: The beam search parameters.
TODO: how does beam search work together with length penalty, frequency
penalty, and stopping criteria, etc.?
""" """
# TODO: how does beam search work together with length penalty,
# frequency, penalty, and stopping criteria, etc.?
beam_width = params.beam_width beam_width = params.beam_width
max_tokens = params.max_tokens max_tokens = params.max_tokens
temperature = params.temperature temperature = params.temperature