Spaces:

nikigoli
/

countgd

Running on T4

nikigoli commited on Jul 6, 2024

Commit

ad6e6cb

verified ·

1 Parent(s): 33ec58a

Update models/GroundingDINO/ops/setup.py

Files changed (1) hide show

models/GroundingDINO/ops/setup.py CHANGED Viewed

@@ -24,7 +24,7 @@ requirements = ["torch", "torchvision"]
 @spaces.GPU
 def get_extensions():
-    this_dir = os.path.dirname(os.path.abspath(__file__))
     extensions_dir = os.path.join(this_dir, "src")
     main_file = glob.glob(os.path.join(extensions_dir, "*.cpp"))

 @spaces.GPU
 def get_extensions():
+    this_dir = "/home/app/models/GroundingDINO/ops" #os.path.dirname(os.path.abspath(__file__))
     extensions_dir = os.path.join(this_dir, "src")
     main_file = glob.glob(os.path.join(extensions_dir, "*.cpp"))