fxmarty 221aaec6ec Ignore non-causal mask in more cases with SDPA (#30138) ...

* update non-causal mask for sdpa

* add test

* update docstrings

* add one more test

* fix cross attention bug

* gentler atol/rtol

2024-06-03 19:08:41 +08:00

..

__init__.py

Move test model folders (#17034)

2022-05-03 14:42:02 +02:00

test_modeling_bert.py

Ignore non-causal mask in more cases with SDPA (#30138)

2024-06-03 19:08:41 +08:00

test_modeling_flax_bert.py

Update all references to canonical models (#29001)

2024-02-16 08:16:58 +01:00

test_modeling_tf_bert.py

Speed up TF tests by reducing hidden layer counts (#24595)

2023-06-30 16:30:33 +01:00

test_tokenization_bert_tf.py

update ruff version (#30932)

2024-05-22 06:40:15 +02:00

test_tokenization_bert.py

update ruff version (#30932)

2024-05-22 06:40:15 +02:00