Skip to content

Commit 8afc01b

Browse files
Chen-0210chenjincong
authored andcommitted
[V1][Minor] fix comments (vllm-project#15392)
Signed-off-by: chenjincong <chenjincong@baidu.com> Signed-off-by: Chen-0210 <chenjincong11@gmail.com> Co-authored-by: chenjincong <chenjincong@baidu.com> Signed-off-by: Mu Huai <tianbowen.tbw@antgroup.com>
1 parent 0532c40 commit 8afc01b

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)