[MM] Pass prefix parameter to MMEncoderAttention (#33674)

Signed-off-by: shen-shanshan <467638484@qq.com>
2026-02-03 22:47:41 +08:00
parent f3d8a34671
commit 5c4f2dd6ef
15 changed files with 58 additions and 11 deletions
--- a/vllm/model_executor/models/step_vl.py
+++ b/vllm/model_executor/models/step_vl.py
@@ -220,7 +220,12 @@ class PerceptionEncoderVisionAttention(nn.Module):
            prefix=f"{prefix}.out_proj",
            disable_tp=use_data_parallel,
        )
-        self.attn = MMEncoderAttention(self.num_heads, self.head_dim, self.scale)
+        self.attn = MMEncoderAttention(
+            self.num_heads,
+            self.head_dim,
+            self.scale,
+            prefix=prefix,
+        )
        self.rope = PerceptionEncoderRope2D(
            dim=self.head_dim,
            max_grid_height=max_grid_height,