tests/test_modeling_flax_clip.py · eb881674f23f4dc90b9dfb03dce9ef918f4f8f9d · chenpangpang / transformers

[Flax] [WIP] allow loading head model with base model weights (#12255) · eb881674

Suraj Patil authored Jun 21, 2021

* boom boom

* remove flax clip example

* allow loading head model with base model weights

* add test

* fix imports

* disable save, load test for clip

* add test_save_load_to_base

eb881674

test_modeling_flax_clip.py 23.1 KB

Replace test_modeling_flax_clip.py