yopo

Runtime error

nikigoli commited on Jul 6, 2024

Commit

74033a0

verified ·

1 Parent(s): 87b1206

Update models/GroundingDINO/ops/setup.py

Files changed (1) hide show

models/GroundingDINO/ops/setup.py CHANGED Viewed

@@ -39,7 +39,7 @@ def get_extensions():
     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

     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