Skip to content

Commit 0893567

Browse files
Chen-0210chenjincong
andauthored
[V1][Minor] fix comments (#15392)
Signed-off-by: chenjincong <[email protected]> Signed-off-by: Chen-0210 <[email protected]> Co-authored-by: chenjincong <[email protected]>
1 parent 8abe69b commit 0893567

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

vllm/v1/sample/sampler.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -137,7 +137,7 @@ def gather_logprobs(
137137
Gather logprobs for topk and sampled/prompt token.
138138
139139
Args:
140-
logits: (num tokens) x (vocab) tensor
140+
logprobs: (num tokens) x (vocab) tensor
141141
num_logprobs: minimum number of logprobs to
142142
retain per token
143143
token_ids: prompt tokens (if prompt logprobs)

0 commit comments

Comments
 (0)