Spaces:
Running
on
Zero
Running
on
Zero
algohunt
commited on
Commit
·
029882a
1
Parent(s):
42c6454
app
Browse files
app.py
CHANGED
|
@@ -49,8 +49,12 @@ WEIGHTS_DIR = os.path.join(os.path.dirname(os.path.abspath(__file__)), 'weights'
|
|
| 49 |
os.makedirs(TMP_DIR, exist_ok=True)
|
| 50 |
os.makedirs(WEIGHTS_DIR, exist_ok=True)
|
| 51 |
|
|
|
|
| 52 |
|
| 53 |
|
|
|
|
|
|
|
|
|
|
| 54 |
|
| 55 |
def cache_weights(weights_dir: str) -> dict:
|
| 56 |
import os
|
|
|
|
| 49 |
os.makedirs(TMP_DIR, exist_ok=True)
|
| 50 |
os.makedirs(WEIGHTS_DIR, exist_ok=True)
|
| 51 |
|
| 52 |
+
is_gpu_available = torch.cuda.is_available()
|
| 53 |
|
| 54 |
|
| 55 |
+
if is_gpu_available:
|
| 56 |
+
print("✅ NVIDIA GPU。")
|
| 57 |
+
|
| 58 |
|
| 59 |
def cache_weights(weights_dir: str) -> dict:
|
| 60 |
import os
|