Fix num_logprobs parameter description in sampler.py (#34451)

Signed-off-by: Zhuohan Li <zhuohan123@gmail.com>
2026-02-12 18:18:03 -08:00
parent fc22cae4ac
commit d707678dfb
1 changed files with 1 additions and 1 deletions
--- a/vllm/v1/sample/sampler.py
+++ b/vllm/v1/sample/sampler.py
@@ -217,7 +217,7 @@ class Sampler(nn.Module):

        Args:
          logprobs: (num tokens) x (vocab) tensor
-          num_logprobs: minimum number of logprobs to
+          num_logprobs: maximum number of logprobs to
                        retain per token
          token_ids: prompt tokens (if prompt logprobs)
                     or sampled tokens (if sampled