Update models/GroundingDINO/ops/setup.py

models/GroundingDINO/ops/setup.py

@@ -11,6 +11,8 @@ import glob
 
 import torch
 
+import spaces
+
 from torch.utils.cpp_extension import CUDA_HOME
 from torch.utils.cpp_extension import CppExtension
 from torch.utils.cpp_extension import CUDAExtension
@@ -20,6 +22,7 @@ from setuptools import setup
 
 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")