We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 8abe69b commit 0893567Copy full SHA for 0893567
vllm/v1/sample/sampler.py
@@ -137,7 +137,7 @@ def gather_logprobs(
137
Gather logprobs for topk and sampled/prompt token.
138
139
Args:
140
- logits: (num tokens) x (vocab) tensor
+ logprobs: (num tokens) x (vocab) tensor
141
num_logprobs: minimum number of logprobs to
142
retain per token
143
token_ids: prompt tokens (if prompt logprobs)
0 commit comments