nikigoli commited on
Commit
28446a4
1 Parent(s): fc2bb80

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