Fixed Deformable Detr typo when loading cuda kernels for MSDA (#29294)

3b8c0536 · Eduardo Pacheco · GitHub · a44d2dc3 · 3b8c0536
Unverified Commit 3b8c0536 authored Feb 26, 2024 by Eduardo Pacheco Committed by GitHub Feb 26, 2024
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

src/transformers/models/deformable_detr/modeling_deformable_detr.py ...ormers/models/deformable_detr/modeling_deformable_detr.py +1 -1

No files found.
--- a/src/transformers/models/deformable_detr/modeling_deformable_detr.py
+++ b/src/transformers/models/deformable_detr/modeling_deformable_detr.py
@@ -60,7 +60,7 @@ def load_cuda_kernels():

    global MultiScaleDeformableAttention

-    root = Path(__file__).resolve().parent.parent.parent / "kernels" / "deta"
+    root = Path(__file__).resolve().parent.parent.parent / "kernels" / "deformable_detr"
    src_files = [
        root / filename
        for filename in [