Forced import _C

models/GroundingDINO/ms_deform_attn.py

@@ -25,12 +25,12 @@ from torch.autograd import Function
 from torch.autograd.function import once_differentiable
 from torch.nn.init import constant_, xavier_uniform_
 
-try:
+#try:
     # from groundingdino import _C
     #import MultiScaleDeformableAttention as _C
-    from functions.ms_deform_attn_func import MSDA as _C
-except:
-    warnings.warn("Failed to load custom C++ ops. Running on CPU mode Only!")
+from functions.ms_deform_attn_func import MSDA as _C
+#except:
+    #warnings.warn("Failed to load custom C++ ops. Running on CPU mode Only!")
 
 
 # helpers