nikigoli commited on
Commit
5e25970
·
verified ·
1 Parent(s): c7135a1

Update models/GroundingDINO/ms_deform_attn.py

Browse files
models/GroundingDINO/ms_deform_attn.py CHANGED
@@ -25,12 +25,12 @@ from torch.autograd import Function
25
  from torch.autograd.function import once_differentiable
26
  from torch.nn.init import constant_, xavier_uniform_
27
 
28
- try:
29
- from groundingdino import _C
30
- import MultiScaleDeformableAttention as _C
31
- #import MultiScaleDeformableAttention as _C
32
- except:
33
- warnings.warn("Failed to load custom C++ ops. Running on CPU mode Only!")
34
 
35
 
36
  # helpers
 
25
  from torch.autograd.function import once_differentiable
26
  from torch.nn.init import constant_, xavier_uniform_
27
 
28
+ #try:
29
+ # from groundingdino import _C
30
+ # import MultiScaleDeformableAttention as _C
31
+ import MultiScaleDeformableAttention as _C
32
+ #except:
33
+ # warnings.warn("Failed to load custom C++ ops. Running on CPU mode Only!")
34
 
35
 
36
  # helpers