PR #29013 DeformableDetrModel support fp16

DeformableDetrModel support fp16 #29013

amyeroberts merged 11 commits into huggingface:main from DonggeunYu:main

Update ms_deform_attn_cuda.cu

95ff15b5

Update ms_deform_attn_cuda.cuh

10d03372

Update modeling_deformable_detr.py

84eaeef7

DonggeunYu commented on 2024-02-14

amyeroberts approved these changes on 2024-02-14

amyeroberts requested a review from

amyeroberts 2 years ago

Update src/transformers/models/deformable_detr/modeling_deformable_de…

7773d5c5

Update modeling_deformable_detr.py

9b3789cd

python utils/check_copies.py --fix_and_overwrite

ad0e2989

Fix dtype missmatch error

743b518c

Update test_modeling_deformable_detr.py

6a7beb9c

Update test_modeling_deformable_detr.py

350cf84a

Update modeling_deformable_detr.py

4379ec76

Update modeling_deformable_detr.py

26578334

amyeroberts approved these changes on 2024-02-15

amyeroberts merged 5b6fa230 into main 2 years ago

Reviewers

amyeroberts

Assignees

No one assigned

Labels

None yet

Milestone

No milestone