src/transformers/modeling_attn_mask_utils.py · 221aaec6ecf7558e4956dadd662d7d3adb22e420 · chenpangpang / transformers

fxmarty authored Jun 03, 2024

* update non-causal mask for sdpa

* add test

* update docstrings

* add one more test

* fix cross attention bug

* gentler atol/rtol

221aaec6

modeling_attn_mask_utils.py 20.8 KB