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