Unverified Commit 0893567d authored by Chen1022's avatar Chen1022 Committed by GitHub
Browse files

[V1][Minor] fix comments (#15392)


Signed-off-by: default avatarchenjincong <chenjincong@baidu.com>
Signed-off-by: default avatarChen-0210 <chenjincong11@gmail.com>
Co-authored-by: default avatarchenjincong <chenjincong@baidu.com>
parent 8abe69b4
......@@ -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)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment