From 8c50d62f5a51799c2ecc1ad25380a5a6dd7c7180 Mon Sep 17 00:00:00 2001 From: ZiTian Zhao Date: Mon, 11 Aug 2025 00:20:00 +0800 Subject: [PATCH] Remove redundant row_indices unsqueeze operation in MiniCPMO (#22528) Signed-off-by: zitian.zhao --- vllm/model_executor/models/minicpmo.py | 1 - 1 file changed, 1 deletion(-) diff --git a/vllm/model_executor/models/minicpmo.py b/vllm/model_executor/models/minicpmo.py index 1ee0a94c37701..e1746695bd5db 100644 --- a/vllm/model_executor/models/minicpmo.py +++ b/vllm/model_executor/models/minicpmo.py @@ -605,7 +605,6 @@ class MiniCPMO(MiniCPMV2_6): max=size) # Create column indices for broadcasting col_indices = torch.arange(size, device=device).unsqueeze(0) - row_indices = row_indices.unsqueeze(1) start_indices = start_indices.unsqueeze(1) end_indices = end_indices.unsqueeze(1) # Vectorized mask creation