riaz.somc/transformers
1
0
Fork 0
mirror of https://github.com/huggingface/transformers.git synced 2025-08-01 02:31:11 +06:00
Code Issues Actions 3 Packages Projects Releases Wiki Activity

[GroundingDino] Adding ms_deform_attn kernels (#30768)

Browse Source 
* Adding ms_deform_attn kernels to GroundingDino

* Pointing to deformable detr kernels
This commit is contained in:
Eduardo Pacheco 2024-05-13 13:34:45 +02:00 committed by GitHub
parent e52741f601
commit 453893ed15
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/transformers/models/grounding_dino/modeling_grounding_dino.py
View File

@ -73,7 +73,7 @@ def load_cuda_kernels():
global MultiScaleDeformableAttention global MultiScaleDeformableAttention
root = Path(__file__).resolve().parent.parent.parent / "kernels" / "grounding_dino" root = Path(__file__).resolve().parent.parent.parent / "kernels" / "deformable_detr"
src_files = [ src_files = [
root / filename root / filename
for filename in [ for filename in [