[MM] Align the prefix of MMEncoderAttention with Attention (#33750)

Signed-off-by: shen-shanshan <467638484@qq.com>
This commit is contained in:
Shanshan Shen
2026-02-04 12:07:30 +08:00
committed by GitHub
parent 4dffc5e044
commit 9fb27dd3b3
17 changed files with 17 additions and 15 deletions

View File

@@ -139,7 +139,7 @@ class EVA2CLIPAttention(nn.Module):
self.num_heads_per_rank,
self.head_dim,
self.scale,
prefix=prefix,
prefix=f"{prefix}.attn",
)
self.output_dropout = torch.nn.Dropout(config.dropout_prob)