tests/test_modeling_flax_common.py · d5a72b6e19e9b594767f1046bc9ceec997cd69a9 · chenpangpang / transformers

Patrick von Platen authored May 26, 2021

* fix_torch_device_generate_test

* remove @

* change dataclasses to flax ones

* fix typo

* fix jitted tests

* fix bert & electra

d5a72b6e

test_modeling_flax_common.py 13.2 KB