transformers
DeviceGuard added to use Deformable Attention more safely on multi-GPU
#32910
Merged
Merged
DeviceGuard added to use Deformable Attention more safely on multi-GPU #32910
amyeroberts
merged
9 commits into
huggingface:main
from
DonggeunYu:main
Update modeling_deformable_detr.py
dc7b6f02
Update src/transformers/models/deformable_detr/modeling_deformable_de…
a7fad545
Merge branch 'huggingface:main' into main
d77ca5e1
Merge branch 'huggingface:main' into main
87172781
Merge branch 'huggingface:main' into main
9da83b6c
Update ms_deform_attn_cuda.cu
19689d16
Update modeling_deformable_detr.py
beccfc95
Update modeling_deformable_detr.py
94747d43
DonggeunYu
changed the title 
amyeroberts
approved these changes
on 2024-08-22
[empty] this is a empty commit
2544d7f4
Login to write a write a comment.
Login via GitHub