Update models/GroundingDINO/ms_deform_attn.py

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:
-    # from groundingdino import _C
-    #import MultiScaleDeformableAttention as _C
-import MultiScaleDeformableAttention as _C
-#except:
-    #warnings.warn("Failed to load custom C++ ops. Running on CPU mode Only!")
+try:
+    from groundingdino import _C
+    import MultiScaleDeformableAttention as _C
+#import MultiScaleDeformableAttention as _C
+except:
+    warnings.warn("Failed to load custom C++ ops. Running on CPU mode Only!")
 
 
 # helpers