Skip to content

Commit 953f0a0

Browse files
alex-jw-brooksadobrzyn
authored andcommitted
[Doc] Move todo out of beam search docstring (vllm-project#17183)
Signed-off-by: Alex-Brooks <[email protected]> Signed-off-by: Agata Dobrzyniewicz <[email protected]>
1 parent 0293f0d commit 953f0a0

File tree

1 file changed

+2
-4
lines changed

1 file changed

+2
-4
lines changed

vllm/entrypoints/llm.py

Lines changed: 2 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -519,11 +519,9 @@ def beam_search(
519519
prompts: A list of prompts. Each prompt can be a string or a list
520520
of token IDs.
521521
params: The beam search parameters.
522-
523-
TODO: how does beam search work together with length penalty, frequency
524-
penalty, and stopping criteria, etc.?
525522
"""
526-
523+
# TODO: how does beam search work together with length penalty,
524+
# frequency, penalty, and stopping criteria, etc.?
527525
beam_width = params.beam_width
528526
max_tokens = params.max_tokens
529527
temperature = params.temperature

0 commit comments

Comments
 (0)