From 157f9c13687e38b89fdeb20ecdbb75baf8153e0f Mon Sep 17 00:00:00 2001 From: Cyrus Leung Date: Fri, 8 Aug 2025 11:21:54 +0800 Subject: [PATCH] Fix pre-commit (#22487) Signed-off-by: DarkLight1337 --- vllm/model_executor/models/minicpmo.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/vllm/model_executor/models/minicpmo.py b/vllm/model_executor/models/minicpmo.py index fd91c7fcc1..1ee0a94c37 100644 --- a/vllm/model_executor/models/minicpmo.py +++ b/vllm/model_executor/models/minicpmo.py @@ -589,7 +589,7 @@ class MiniCPMO(MiniCPMV2_6): ret = torch.zeros(size, size, device=device, dtype=torch.bool) # Vectorized computation of row indices and chunk boundaries row_indices = torch.arange(size, device=device) - chunk_indices = row_indices // chunk_size + chunk_indices = row_indices // chunk_size if num_left_chunks < 0: # If num_left_chunks < 0, start is always 0 for all rows start_indices = torch.zeros_like(row_indices) @@ -597,7 +597,7 @@ class MiniCPMO(MiniCPMV2_6): # Compute start indices vectorially start_chunk_indices = torch.clamp(chunk_indices - num_left_chunks, min=0) - start_indices = start_chunk_indices * chunk_size + start_indices = start_chunk_indices * chunk_size # Compute ending indices vectorially end_chunk_indices = chunk_indices + 1 end_indices = torch.clamp(end_chunk_indices * chunk_size +