Spaces:
Running
on
T4
Running
on
T4
Update models/GroundingDINO/ops/setup.py
Browse files
models/GroundingDINO/ops/setup.py
CHANGED
@@ -39,7 +39,7 @@ def get_extensions():
|
|
39 |
|
40 |
|
41 |
print("inside get_extensions")
|
42 |
-
print(torch.cuda.is_available)
|
43 |
print(CUDA_HOME)
|
44 |
if torch.cuda.is_available() and CUDA_HOME is not None:
|
45 |
extension = CUDAExtension
|
|
|
39 |
|
40 |
|
41 |
print("inside get_extensions")
|
42 |
+
print(torch.cuda.is_available())
|
43 |
print(CUDA_HOME)
|
44 |
if torch.cuda.is_available() and CUDA_HOME is not None:
|
45 |
extension = CUDAExtension
|