Fix bigbird random attention #21023
switch np.random.permutation to jax.random.permuation
461476df
Merge branch 'main' of https://github.com/huggingface/transformers in…
69804b71
remove comments
2fcb68b7
remove leftover comment
8b0c8750
skip similarity tests
dceb7eac
Merge branch 'fix_bigbird_random_attention' of https://github.com/Bea…
843a6c05
modify indices_prng_key usage, add deterministic behaviour
fa55da25
update style
de25471c
remove unused import
2a8d660f
remove copy statement since classes are not identical
3978cf6e
Merge branch 'main' of https://github.com/huggingface/transformers in…
e3676b90
remove numpy import
1c37ccc1
Merge branch 'main' of https://github.com/huggingface/transformers in…
e4799f1d
revert removing copied from statements
88688ff6
make style from copied
fcadb2fd
remove copied from statement
b2754c59
update copied from statement to include only np.ndarry
1318438c
Merge branch 'main' of https://github.com/huggingface/transformers in…
597c1d60
add deterministic args, unittestskip equivalence tests
bd735361
sgugger
merged
88399476
into main 3 years ago
Assignees
No one assigned
Login to write a write a comment.
Login via GitHub