Billy Cao 1f9f57ab4c Fix dtype casting in swinv2 and swinv2sr to allow non-FP32 inference (#31589) ... * 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>		2024-06-26 18:46:48 +01:00
..
__init__.py	Add swin transformer v2 (#17469)	2022-07-27 11:14:47 -04:00
test_modeling_swinv2.py	Fix dtype casting in swinv2 and swinv2sr to allow non-FP32 inference (#31589)	2024-06-26 18:46:48 +01:00