File tree Expand file tree Collapse file tree 1 file changed +3
-3
lines changed
vllm/model_executor/layers Expand file tree Collapse file tree 1 file changed +3
-3
lines changed Original file line number Diff line number Diff line change @@ -283,14 +283,14 @@ def _get_accepted(
283
283
batch_size , k , _ = draft_probs .shape
284
284
batch_indices = torch .arange (batch_size ,
285
285
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 )
287
287
288
288
# shape [batch_size, k]
289
- selected_draft_probs = draft_probs [batch_indices , probs_indicies ,
289
+ selected_draft_probs = draft_probs [batch_indices , probs_indices ,
290
290
draft_token_ids ]
291
291
292
292
# shape [batch_size, k]
293
- selected_target_probs = target_probs [batch_indices , probs_indicies ,
293
+ selected_target_probs = target_probs [batch_indices , probs_indices ,
294
294
draft_token_ids ]
295
295
296
296
uniform_rand = self ._create_uniform_samples (seeded_seqs , batch_size ,
You can’t perform that action at this time.
0 commit comments