Spaces:

sunnychenxiwang
/

EasyDetect

Sleeping

sunnychenxiwang commited on Jan 17, 2024

Commit

4128a10

verified ·

1 Parent(s): 912c01d

Update pipeline/GroundingDINO/groundingdino/models/GroundingDINO/ms_deform_attn.py

Files changed (1) hide show

pipeline/GroundingDINO/groundingdino/models/GroundingDINO/ms_deform_attn.py CHANGED Viewed

@@ -26,7 +26,7 @@ from torch.autograd.function import once_differentiable
 from torch.nn.init import constant_, xavier_uniform_
 try:
-    from groundingdino import _C
 except:
     warnings.warn("Failed to load custom C++ ops. Running on CPU mode Only!")

 from torch.nn.init import constant_, xavier_uniform_
 try:
+    from pipeline.GroundingDINO.groundingdino import _C
 except:
     warnings.warn("Failed to load custom C++ ops. Running on CPU mode Only!")