From 3c8694eabe60e37fbbf2e71aa1414f1370b5014b Mon Sep 17 00:00:00 2001 From: Ximingwang-09 <72070413+Ximingwang-09@users.noreply.github.com> Date: Wed, 11 Jun 2025 18:36:04 +0800 Subject: [PATCH] Fix some typo (#19475) Signed-off-by: ximing.wxm Co-authored-by: ximing.wxm --- vllm/model_executor/layers/rejection_sampler.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/vllm/model_executor/layers/rejection_sampler.py b/vllm/model_executor/layers/rejection_sampler.py index a6e58a77d42c..db68f18726d3 100644 --- a/vllm/model_executor/layers/rejection_sampler.py +++ b/vllm/model_executor/layers/rejection_sampler.py @@ -283,14 +283,14 @@ class RejectionSampler(SpecDecodeStochasticBaseSampler): batch_size, k, _ = draft_probs.shape batch_indices = torch.arange(batch_size, device=target_probs.device)[:, None] - probs_indicies = torch.arange(k, device=target_probs.device) + probs_indices = torch.arange(k, device=target_probs.device) # shape [batch_size, k] - selected_draft_probs = draft_probs[batch_indices, probs_indicies, + selected_draft_probs = draft_probs[batch_indices, probs_indices, draft_token_ids] # shape [batch_size, k] - selected_target_probs = target_probs[batch_indices, probs_indicies, + selected_target_probs = target_probs[batch_indices, probs_indices, draft_token_ids] uniform_rand = self._create_uniform_samples(seeded_seqs, batch_size,