tests/test_modeling_flax_common.py · 4e9a6796c7f2b26f5225f4ed3e8f74071947c2d0 · chenpangpang / transformers · GitLab

Find file Blame History Permalink

[Flax] Fix flax test save pretrained (#12256) · 4e9a6796
Patrick von Platen authored Jun 21, 2021
```
* fix_torch_device_generate_test

* remove @

* fix flax save pretrained test
```
4e9a6796

test_modeling_flax_common.py 21.9 KB