Spaces:
Running
on
T4
Running
on
T4
Update ms_deform_attn.py in main/models/GroundingDINO/ with different MSDA import
Browse files
models/GroundingDINO/ms_deform_attn.py
CHANGED
@@ -27,7 +27,8 @@ from torch.nn.init import constant_, xavier_uniform_
|
|
27 |
|
28 |
try:
|
29 |
# from groundingdino import _C
|
30 |
-
import MultiScaleDeformableAttention as _C
|
|
|
31 |
except:
|
32 |
warnings.warn("Failed to load custom C++ ops. Running on CPU mode Only!")
|
33 |
|
|
|
27 |
|
28 |
try:
|
29 |
# from groundingdino import _C
|
30 |
+
#import MultiScaleDeformableAttention as _C
|
31 |
+
from functions.ms_deform_attn_func import MSDA as _C
|
32 |
except:
|
33 |
warnings.warn("Failed to load custom C++ ops. Running on CPU mode Only!")
|
34 |
|