Spaces:

nikigoli
/

countgd

Running on T4

nikigoli commited on Jul 15

Commit

0d11f1e

•

1 Parent(s): a657eeb

Update models/GroundingDINO/ms_deform_attn.py

Files changed (1) hide show

models/GroundingDINO/ms_deform_attn.py CHANGED Viewed

@@ -28,7 +28,7 @@ from torch.nn.init import constant_, xavier_uniform_
 #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!")

 #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!")