tests/test_modeling_flax_albert.py · 98e409abb3a91e4de2c27e1c729ba9fdaa74d3df · chenpangpang / transformers

Kamal Raj authored Aug 30, 2021

* albert flax

* year -> 2021

* docstring updated for flax

* removed head_mask

* removed from_pt

* removed passing attention_mask to embedding layer

98e409ab

test_modeling_flax_albert.py 5.83 KB

Replace test_modeling_flax_albert.py