transformers
1dbd9d36 - DeviceGuard added to use Deformable Attention more safely on multi-GPU (#32910)

Commit
1 year ago
DeviceGuard added to use Deformable Attention more safely on multi-GPU (#32910) * Update modeling_deformable_detr.py * Update src/transformers/models/deformable_detr/modeling_deformable_detr.py Co-authored-by: amyeroberts <22614925+amyeroberts@users.noreply.github.com> * Update ms_deform_attn_cuda.cu * Update modeling_deformable_detr.py * Update modeling_deformable_detr.py * [empty] this is a empty commit --------- Co-authored-by: amyeroberts <22614925+amyeroberts@users.noreply.github.com>
Author
Parents
Loading