tests/test_modeling_flax_common.py · aebca696afb511013d71fb6da5c162d945fb31d4 · chenpangpang / transformers

Fix missing output_attentions in PT/Flax equivalence test (#16271) · aebca696

Yih-Dar authored Mar 29, 2022



* fix - set output_attentions to True

* Update tests/test_modeling_flax_common.py

* update for has_attentions

* overwrite check_outputs in FlaxBigBirdModelTest
Co-authored-by: ydshieh <ydshieh@users.noreply.github.com>
Co-authored-by: Suraj Patil <surajp815@gmail.com>

aebca696

test_modeling_flax_common.py 42.4 KB

Replace test_modeling_flax_common.py