tests/test_modeling_flax_gpt2.py · 2bef3433e5fa86345331ac3d856a63f6931f70bd · chenpangpang / transformers · GitLab

Find file Blame History Permalink

[Flax] Correct all return tensors to numpy (#13307) · 2bef3433
Patrick von Platen authored Aug 27, 2021
```
* fix_torch_device_generate_test

* remove @

* finish find and replace
```
2bef3433

test_modeling_flax_gpt2.py 14.6 KB