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

Signed-off-by: shen-shanshan <467638484@qq.com>
This commit is contained in:
Shanshan Shen
2026-02-03 22:47:41 +08:00
committed by GitHub
parent f3d8a34671
commit 5c4f2dd6ef
15 changed files with 58 additions and 11 deletions

View File

@@ -136,7 +136,10 @@ class EVA2CLIPAttention(nn.Module):
)
self.attn = MMEncoderAttention(
self.num_heads_per_rank, self.head_dim, self.scale
self.num_heads_per_rank,
self.head_dim,
self.scale,
prefix=prefix,
)
self.output_dropout = torch.nn.Dropout(config.dropout_prob)