Spaces:
Running
on
T4
Running
on
T4
Commit
·
da1e5da
1
Parent(s):
2f6473f
debug gpu
Browse files- app.py +1 -1
- settings.py +3 -2
app.py
CHANGED
@@ -8,7 +8,7 @@ from app_canny import create_demo as create_demo_canny
|
|
8 |
from model import Model
|
9 |
from settings import ALLOW_CHANGING_BASE_MODEL, DEFAULT_MODEL_ID, SHOW_DUPLICATE_BUTTON
|
10 |
|
11 |
-
DESCRIPTION = "# Material Authoring Demo v0.1"
|
12 |
|
13 |
if not torch.cuda.is_available():
|
14 |
DESCRIPTION += "\n<p>Running on CPU 🥶 This demo does not work on CPU.</p>"
|
|
|
8 |
from model import Model
|
9 |
from settings import ALLOW_CHANGING_BASE_MODEL, DEFAULT_MODEL_ID, SHOW_DUPLICATE_BUTTON
|
10 |
|
11 |
+
DESCRIPTION = "# Material Authoring Demo v0.1. Under Construction"
|
12 |
|
13 |
if not torch.cuda.is_available():
|
14 |
DESCRIPTION += "\n<p>Running on CPU 🥶 This demo does not work on CPU.</p>"
|
settings.py
CHANGED
@@ -15,5 +15,6 @@ SHOW_DUPLICATE_BUTTON = os.getenv("SHOW_DUPLICATE_BUTTON") == "1"
|
|
15 |
MAX_SEED = np.iinfo(np.int32).max
|
16 |
|
17 |
# setup CUDA
|
18 |
-
|
19 |
-
|
|
|
|
15 |
MAX_SEED = np.iinfo(np.int32).max
|
16 |
|
17 |
# setup CUDA
|
18 |
+
# disable the following when deployting to hugging face
|
19 |
+
# if os.getenv("CUDA_VISIBLE_DEVICES") is None:
|
20 |
+
# os.environ["CUDA_VISIBLE_DEVICES"] = "7"
|