tests/test_modeling_flax_common.py · 007be9e402918b881ee89508512e1b742d818b45 · chenpangpang / transformers · GitLab

Find file Blame History Permalink

[Flax] Fix flax pt equivalence tests (#12154) · 007be9e4
Patrick von Platen authored Jun 14, 2021
```
* fix_torch_device_generate_test

* remove @

* upload
```
007be9e4

test_modeling_flax_common.py 19.9 KB