Skip to content

[Backport] Allow setting FlashAttention's causal mask #6625

[Backport] Allow setting FlashAttention's causal mask

[Backport] Allow setting FlashAttention's causal mask #6625

Annotations

1 warning

GPU tests  /  test (python_tests, xla_op1)

succeeded Mar 27, 2024 in 1h 11m 45s