修复一个bug

2 years ago · 0d1c6d05a7
parent 4080c92355
commit 0d1c6d05a7
1 changed files with 1 additions and 1 deletions
--- a/setup.py
+++ b/setup.py
@ -55,7 +55,7 @@ torch_ver = [int(x) for x in torch.__version__.split(".")[:2]]

 def get_extensions():
    this_dir = os.path.dirname(os.path.abspath(__file__))
-    extensions_dir = os.path.join(this_dir, "groundingdino", "models", "GroundingDINO", "csrc","datasets")
+    extensions_dir = os.path.join(this_dir, "groundingdino", "models", "GroundingDINO", "csrc")

    main_source = os.path.join(extensions_dir, "vision.cpp")
    sources = glob.glob(os.path.join(extensions_dir, "**", "*.cpp"))