[MM] Pass prefix parameter to MMEncoderAttention (#33674)
Signed-off-by: shen-shanshan <467638484@qq.com>
This commit is contained in:
@@ -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)
|
||||
|
||||
|
||||
Reference in New Issue
Block a user