tests/models/bert/test_modeling_bert.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

test_modeling_bert.py 31.2 KB