Remove redundant row_indices unsqueeze operation in MiniCPMO (#22528)

Signed-off-by: zitian.zhao <zitian.zhao@tencentmusic.com>
2025-08-11 00:20:00 +08:00
parent b4e2916721
commit 8c50d62f5a
1 changed files with 0 additions and 1 deletions
--- 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