Skip to content

Commit 76bf018

Browse files
authored
Update validate_installation.py (#16)
1 parent 88e8cae commit 76bf018

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

scripts/validate_installation.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -80,8 +80,10 @@ def check_cuda_support():
8080
print(f" Device {i}: {torch.cuda.get_device_name(i)}")
8181
else:
8282
print(f"\n⚠ CUDA is not available (CPU-only mode)")
83+
return True
8384
except ImportError:
8485
print(f"\n⚠ PyTorch not installed - cannot check CUDA support")
86+
return False
8587

8688

8789
def check_croco_rope():
@@ -144,7 +146,6 @@ def check_file_structure():
144146
"uniception/models/info_sharing",
145147
"uniception/models/prediction_heads",
146148
"scripts",
147-
"tests",
148149
]
149150

150151
missing_dirs = []

0 commit comments

Comments
 (0)