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

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

View File

@ -519,11 +519,9 @@ class LLM:
prompts: A list of prompts. Each prompt can be a string or a list
of token IDs.
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
max_tokens = params.max_tokens
temperature = params.temperature