From 0893567db9350f4760434ae9e140881c58c5f1d7 Mon Sep 17 00:00:00 2001 From: Chen1022 <112855051+Chen-0210@users.noreply.github.com> Date: Mon, 24 Mar 2025 23:45:32 +0800 Subject: [PATCH] [V1][Minor] fix comments (#15392) Signed-off-by: chenjincong Signed-off-by: Chen-0210 Co-authored-by: chenjincong --- vllm/v1/sample/sampler.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/vllm/v1/sample/sampler.py b/vllm/v1/sample/sampler.py index abff7c1c2652b..397a049dc2543 100644 --- a/vllm/v1/sample/sampler.py +++ b/vllm/v1/sample/sampler.py @@ -137,7 +137,7 @@ class Sampler(nn.Module): Gather logprobs for topk and sampled/prompt token. Args: - logits: (num tokens) x (vocab) tensor + logprobs: (num tokens) x (vocab) tensor num_logprobs: minimum number of logprobs to retain per token token_ids: prompt tokens (if prompt logprobs)