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 Swin2SR (#19784)	2022-12-16 16:24:01 +01:00
test_image_processing_swin2sr.py	Add ViTImageProcessorFast to tests (#31424)	2024-06-25 13:36:58 +01:00
test_modeling_swin2sr.py	Fix dtype casting in swinv2 and swinv2sr to allow non-FP32 inference (#31589)	2024-06-26 18:46:48 +01:00