tests/test_modeling_flax_common.py · 90178b0cefe94fef258a39cff5019b5ec150597b · chenpangpang / transformers

Add option to load a pretrained model with mismatched shapes (#12664) · 90178b0c

Sylvain Gugger authored Jul 13, 2021



* Add option to load a pretrained model with mismatched shapes

* Fail at loading when mismatched shapes in Flax

* Fix tests

* Update src/transformers/modeling_flax_utils.py
Co-authored-by: Patrick von Platen <patrick.v.platen@gmail.com>

* Address review comments
Co-authored-by: Patrick von Platen <patrick.v.platen@gmail.com>

90178b0c

test_modeling_flax_common.py 25.9 KB

Replace test_modeling_flax_common.py