tests/poolformer/test_modeling_poolformer.py · 8d83ebdf189918f2aca25fc90b662836555862d5 · chenpangpang / transformers

[Tests] Add attentions_option to ModelTesterMixin (#15909) · 8d83ebdf

NielsRogge authored Mar 10, 2022



* Add attentions_option to common tester

* Fix tests, apply suggestion

* Apply suggestion from code review
Co-authored-by: Niels Rogge <nielsrogge@Nielss-MacBook-Pro.local>

8d83ebdf

test_modeling_poolformer.py 8.87 KB

Replace test_modeling_poolformer.py