Bartosz Szmelczynski 88399476c3 Fix bigbird random attention (#21023) ... * switch np.random.permutation to jax.random.permuation * remove comments * remove leftover comment * skip similarity tests * modify indices_prng_key usage, add deterministic behaviour * update style * remove unused import * remove copy statement since classes are not identical * remove numpy import * revert removing copied from statements * make style from copied * remove copied from statement * update copied from statement to include only np.ndarry * add deterministic args, unittestskip equivalence tests		2023-04-27 13:52:28 -04:00
..
__init__.py	Move test model folders (#17034)	2022-05-03 14:42:02 +02:00
test_modeling_big_bird.py	🔥Rework pipeline testing by removing PipelineTestCaseMeta 🚀 (#21516)	2023-02-28 19:40:57 +01:00
test_modeling_flax_big_bird.py	Fix bigbird random attention (#21023)	2023-04-27 13:52:28 -04:00
test_tokenization_big_bird.py	Update quality tooling for formatting (#21480)	2023-02-06 18:10:56 -05:00