Skip to content

Commit 157f9c1

Browse files
Fix pre-commit (#22487)
Signed-off-by: DarkLight1337 <[email protected]>
1 parent 6f28791 commit 157f9c1

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

vllm/model_executor/models/minicpmo.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -589,15 +589,15 @@ def subsequent_chunk_mask(
589589
ret = torch.zeros(size, size, device=device, dtype=torch.bool)
590590
# Vectorized computation of row indices and chunk boundaries
591591
row_indices = torch.arange(size, device=device)
592-
chunk_indices = row_indices // chunk_size
592+
chunk_indices = row_indices // chunk_size
593593
if num_left_chunks < 0:
594594
# If num_left_chunks < 0, start is always 0 for all rows
595595
start_indices = torch.zeros_like(row_indices)
596596
else:
597597
# Compute start indices vectorially
598598
start_chunk_indices = torch.clamp(chunk_indices - num_left_chunks,
599599
min=0)
600-
start_indices = start_chunk_indices * chunk_size
600+
start_indices = start_chunk_indices * chunk_size
601601
# Compute ending indices vectorially
602602
end_chunk_indices = chunk_indices + 1
603603
end_indices = torch.clamp(end_chunk_indices * chunk_size +

0 commit comments

Comments
 (0)