[Docs] Fix warnings in mkdocs build (continued) (#23743)

Signed-off-by: Zerohertz <ohg3417@gmail.com>
Signed-off-by: Hyogeun Oh (오효근) <ohg3417@gmail.com>
This commit is contained in:
Hyogeun Oh (오효근)
2025-08-28 02:17:29 +09:00
committed by GitHub
parent dd58932280
commit 4e4d017b6f
26 changed files with 56 additions and 50 deletions

View File

@@ -68,7 +68,7 @@ class RejectionSampler(nn.Module):
different requests are flattened into a single tensor because
this is the shape of the output logits.
NOTE: `target_logits` can be updated in place to save memory.
bonus_token_ids_tensor (torch.Tensor):
bonus_token_ids (torch.Tensor):
A tensor containing bonus tokens. Shape is [batch_size, 1].
Bonus tokens are added to the end of the sequence if all
proposed tokens are accepted. We generate the bonus tokens