Spaces:

nikigoli
/

countgd

Running on T4

nikigoli commited on Jul 6, 2024

Commit

9705e1a

•

1 Parent(s): ad6e6cb

Update models/GroundingDINO/ops/setup.py

Files changed (1) hide show

models/GroundingDINO/ops/setup.py CHANGED Viewed

@@ -22,9 +22,9 @@ from setuptools import setup
 requirements = ["torch", "torchvision"]
-@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"))

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