Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -12,7 +12,7 @@ import os
|
|
12 |
import warnings
|
13 |
warnings.filterwarnings("ignore")
|
14 |
|
15 |
-
if ("0"
|
16 |
os.environ["CUDA_DEVICE_ORDER"] = "PCI_BUS_ID"
|
17 |
os.environ["CUDA_VISIBLE_DEVICES"] = ""
|
18 |
|
|
|
12 |
import warnings
|
13 |
warnings.filterwarnings("ignore")
|
14 |
|
15 |
+
if ("0" != "mycustom"):
|
16 |
os.environ["CUDA_DEVICE_ORDER"] = "PCI_BUS_ID"
|
17 |
os.environ["CUDA_VISIBLE_DEVICES"] = ""
|
18 |
|