fix bug

deeplink_ext/internlm_ops/mha/mha.py

@@ -49,7 +49,6 @@ def forward(self, qkv, causal=None, cu_seqlens=None, max_seqlen=None):
                 self.dropout_p if self.training else 0.0,
                 self.softmax_scale,
                 causal if causal is not None else self.causal,
-                False,
             )
         # else:
         #     # unpadded
@@ -89,7 +88,6 @@ def forward(
                 self.dropout_p if self.training else 0.0,
                 self.softmax_scale,
                 causal if causal is not None else self.causal,
-                False,
             )
         # else:
         #     # unpadded