tests/test_modeling_flax_common.py · fcb4f11c9232cee2adce8140a3a7689578ea97de · chenpangpang / transformers

Add a main_input_name attribute to all models (#14803) · 33f36c86

Sylvain Gugger authored Dec 20, 2021



* Add a main_input_name attribute to all models

* Fix tests

* Wtf Vs Code?

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

* Style

* Fix copies
Co-authored-by: Patrick von Platen <patrick.v.platen@gmail.com>

33f36c86

test_modeling_flax_common.py 39.3 KB

Replace test_modeling_flax_common.py