tests/models/kosmos2/test_modeling_kosmos2.py · a6e0d5a219c3a05171e695d375b1e35cf2915d71 · chenpangpang / transformers

Fix safetensors failing tests (#27231) · 443bf5e9

Lysandre Debut authored Nov 02, 2023



* Fix Kosmos2

* Fix ProphetNet

* Fix MarianMT

* Fix M4T

* XLM ProphetNet

* ProphetNet fix

* XLM ProphetNet

* Final M4T fixes

* Tied weights keys

* Revert M4T changes

* Apply suggestions from code review
Co-authored-by: amyeroberts <22614925+amyeroberts@users.noreply.github.com>

---------
Co-authored-by: amyeroberts <22614925+amyeroberts@users.noreply.github.com>

443bf5e9

test_modeling_kosmos2.py 32.8 KB

Replace test_modeling_kosmos2.py