riaz.somc/transformers
1
0
Fork 0
mirror of https://github.com/huggingface/transformers.git synced 2025-07-03 12:50:06 +06:00
Code Issues Actions 28 Packages Projects Releases Wiki Activity

Fixes: Corrects file path for CUDA kernels (#37438)

Browse Source 
Corrects the file path used to locate the CUDA kernels
for the Deformable Attention module. This ensures that
the kernels are loaded correctly, resolving potential
errors during module initialization and usage.
This commit is contained in:
Donggeun Yu 2025-04-11 17:41:46 +09:00 committed by GitHub
parent c7064cdba1
commit 931126b929
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/deprecated/deta/modeling_deta.py
View File

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