[Misc][MM] Remove redundant code in CLIPAttention (#43046)

Signed-off-by: shen-shanshan <467638484@qq.com>
2026-05-19 08:43:16 +00:00
parent 07beaed842
commit ef54a4d604
1 changed files with 6 additions and 14 deletions
@@ -397,20 +397,12 @@ class CLIPAttention(nn.Module):
        )
        self.num_heads_per_partition = divide(self.num_heads, self.tp_size)

-        if attn_cls == MMEncoderAttention:
-            self.attn = attn_cls(
-                self.num_heads_per_partition,
-                self.head_dim,
-                self.scale,
-                prefix=f"{prefix}.attn",
-            )
-        else:
-            self.attn = attn_cls(
-                self.num_heads_per_partition,
-                self.head_dim,
-                self.scale,
-                prefix=f"{prefix}.attn",
-            )
+        self.attn = attn_cls(
+            self.num_heads_per_partition,
+            self.head_dim,
+            self.scale,
+            prefix=f"{prefix}.attn",
+        )

    def forward(
        self,