[MM] Align the prefix of MMEncoderAttention with Attention (#33750)
Signed-off-by: shen-shanshan <467638484@qq.com>
This commit is contained in:
@@ -224,7 +224,7 @@ class PerceptionEncoderVisionAttention(nn.Module):
|
||||
self.num_heads,
|
||||
self.head_dim,
|
||||
self.scale,
|
||||
prefix=prefix,
|
||||
prefix=f"{prefix}.attn",
|
||||
)
|
||||
self.rope = PerceptionEncoderRope2D(
|
||||
dim=self.head_dim,
|
||||
|
||||
Reference in New Issue
Block a user