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): ...@@ -137,7 +137,7 @@ class Sampler(nn.Module):
Gather logprobs for topk and sampled/prompt token. Gather logprobs for topk and sampled/prompt token.
Args: Args:
logits: (num tokens) x (vocab) tensor logprobs: (num tokens) x (vocab) tensor
num_logprobs: minimum number of logprobs to num_logprobs: minimum number of logprobs to
retain per token retain per token
token_ids: prompt tokens (if prompt logprobs) 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