Update sam2/modeling/sam/transformer.py

sam2/modeling/sam/transformer.py

@@ -322,8 +322,7 @@ class RoPEAttention(Attention):
         v_bshd = rearrange(v_hsd, "b h s d -> b s h d")
     
         out = flash_attn_interface.flash_attn_func(
-            q_bshd, k_bshd, v_bshd,
-            dropout_p=self.dropout_p if self.training else 0.0
+            q_bshd, k_bshd, v_bshd
         )  # (B, S, H, D)
     
         out = rearrange(out, "b s h d -> b s (h d)")