tests/models/swin2sr/test_modeling_swin2sr.py · 1f9f57ab4c8c30964360a2ba697c339f6d31f03f · chenpangpang / transformers

"tests/models/donut/test_modeling_donut_swin.py" did not exist on "2e7e4280aa6f380a4e3afad6524295a17901c56c"

Fix dtype casting in swinv2 and swinv2sr to allow non-FP32 inference (#31589) · 1f9f57ab

Billy Cao authored Jun 27, 2024



* Fix dtype casting in modeling_swin2sr to allow non-FP32 inference

* Fix formattting

* Fix for swinv2 too

* Update src/transformers/models/swin2sr/modeling_swin2sr.py
Co-authored-by: amyeroberts <22614925+amyeroberts@users.noreply.github.com>

* Update src/transformers/models/swinv2/modeling_swinv2.py
Co-authored-by: amyeroberts <22614925+amyeroberts@users.noreply.github.com>

* Add FP16 tests for swin2sr and swinv2

* [run_slow] swin2sr, swinv2

* [run_slow] swin2sr, swinv2

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

1f9f57ab

test_modeling_swin2sr.py 13.3 KB

Replace test_modeling_swin2sr.py