Sylvain Gugger 90178b0cef Add option to load a pretrained model with mismatched shapes (#12664) ...

* 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>

2021-07-13 10:15:15 -04:00

26 KiB

Raw Blame History

View Raw