Spaces:

nikigoli
/

countgd

Running on T4

nikigoli commited on Jul 6

Commit

87b1206

•

1 Parent(s): 9705e1a

Update models/GroundingDINO/ops/setup.py

Files changed (1) hide show

models/GroundingDINO/ops/setup.py CHANGED Viewed

@@ -38,6 +38,9 @@ def get_extensions():
     if torch.cuda.is_available() and CUDA_HOME is not None:
         extension = CUDAExtension
         sources += source_cuda

+    print("inside get_extensions")
+    print(torch.cuda.is_available)
+    print(CUDA_HOME)
     if torch.cuda.is_available() and CUDA_HOME is not None:
         extension = CUDAExtension
         sources += source_cuda