transformers
Fix dtype casting in swinv2 and swinv2sr to allow non-FP32 inference
#31589
Merged
Merged
Fix dtype casting in swinv2 and swinv2sr to allow non-FP32 inference #31589
amyeroberts
merged
8 commits into
huggingface:main
from
aliencaocao:fix-swin2sr-dtype
Fix dtype casting in modeling_swin2sr to allow non-FP32 inference
d87a5e23
Fix formattting
631174c7
Fix for swinv2 too
2b83074d
aliencaocao
changed the title 
Update src/transformers/models/swin2sr/modeling_swin2sr.py
bfe3ff8c
Update src/transformers/models/swinv2/modeling_swinv2.py
3ddf1e97
Add FP16 tests for swin2sr and swinv2
606f1d35
amyeroberts
approved these changes
on 2024-06-26
[run_slow] swin2sr, swinv2
1a2a92e7
[run_slow] swin2sr, swinv2
266a6e13
Login to write a write a comment.
Login via GitHub