Skip to content

Commit 8c9d5d3

Browse files
committed
fix
Signed-off-by: Icey <1790571317@qq.com>
1 parent a7ea166 commit 8c9d5d3

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

vllm_ascend/spec_decode/eagle_proposer.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -432,7 +432,8 @@ def _propose(
432432
num_reqs=batch_size,
433433
num_actual_tokens=num_tokens,
434434
actual_seq_lengths_q=self.runner.actual_seq_lengths_q,
435-
block_table_tensor=self.runner.input_batch.block_table[0].get_device_tensor(),
435+
block_table_tensor=self.runner.input_batch.block_table[0].
436+
get_device_tensor(),
436437
slot_mapping_cpu=target_slot_mapping,
437438
positions=target_positions,
438439
attn_mask=self.runner.attn_mask,

0 commit comments

Comments
 (0)