tests/test_modeling_tf_common.py · 91e1f24ef3efd967871b2b81658db08c648275a6 · chenpangpang / transformers

CLI: convert sharded PT models (#17959) · 91e1f24e

Joao Gante authored Jun 30, 2022

* sharded conversion; add flag to control max hidden error

* better hidden name matching

* Add test: load TF from PT shards

* fix test (PT data must be local)

91e1f24e

test_modeling_tf_common.py 99.3 KB

Replace test_modeling_tf_common.py