Spaces:
Running
on
T4
Running
on
T4
sparkleman
commited on
Commit
·
8af9256
1
Parent(s):
dac6105
UPDATE
Browse files
app.py
CHANGED
|
@@ -37,7 +37,7 @@ import torch
|
|
| 37 |
|
| 38 |
if "cuda" in CONFIG.STRATEGY.lower() and not torch.cuda.is_available():
|
| 39 |
logger.info(f"CUDA not found, fall back to cpu")
|
| 40 |
-
CONFIG.STRATEGY = "cpu"
|
| 41 |
|
| 42 |
if "cuda" in CONFIG.STRATEGY.lower():
|
| 43 |
from pynvml import *
|
|
|
|
| 37 |
|
| 38 |
if "cuda" in CONFIG.STRATEGY.lower() and not torch.cuda.is_available():
|
| 39 |
logger.info(f"CUDA not found, fall back to cpu")
|
| 40 |
+
CONFIG.STRATEGY = "cpu fp16"
|
| 41 |
|
| 42 |
if "cuda" in CONFIG.STRATEGY.lower():
|
| 43 |
from pynvml import *
|