examples/flax/language-modeling/run_mlm_flax.py · 0903fc80b5fd17ff5b9350fc497a3ed381800656 · chenpangpang / transformers · GitLab

"model.properties" did not exist on "e012f86c91c9a916f0adca83ecb17da4d93a1e45"

Find file Blame History Permalink

[JAX] Replace all jax.tree_* calls with jax.tree_util.tree_* (#18361) · e6f221c8
Sanchit Gandhi authored Sep 09, 2022
```
* [JAX] Replace all jax.tree_* calls with jax.tree_util.tree_*

* fix double tree_util
```
e6f221c8

run_mlm_flax.py 38 KB