Skip to content

Commit 3c8694e

Browse files
Ximingwang-09ximing.wxm
andauthored
Fix some typo (#19475)
Signed-off-by: ximing.wxm <ximing.wxm@antgroup.com> Co-authored-by: ximing.wxm <ximing.wxm@antgroup.com>
1 parent 7484e1f commit 3c8694e

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

vllm/model_executor/layers/rejection_sampler.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -283,14 +283,14 @@ def _get_accepted(
283283
batch_size, k, _ = draft_probs.shape
284284
batch_indices = torch.arange(batch_size,
285285
device=target_probs.device)[:, None]
286-
probs_indicies = torch.arange(k, device=target_probs.device)
286+
probs_indices = torch.arange(k, device=target_probs.device)
287287

288288
# shape [batch_size, k]
289-
selected_draft_probs = draft_probs[batch_indices, probs_indicies,
289+
selected_draft_probs = draft_probs[batch_indices, probs_indices,
290290
draft_token_ids]
291291

292292
# shape [batch_size, k]
293-
selected_target_probs = target_probs[batch_indices, probs_indicies,
293+
selected_target_probs = target_probs[batch_indices, probs_indices,
294294
draft_token_ids]
295295

296296
uniform_rand = self._create_uniform_samples(seeded_seqs, batch_size,

0 commit comments

Comments
 (0)