Spaces:
Runtime error
Runtime error
Commit
·
bb0b433
1
Parent(s):
9766f45
update
Browse files
app.py
CHANGED
@@ -61,7 +61,13 @@ if opt.resume is not None:
|
|
61 |
else:
|
62 |
print(f'[WARN] model randomly initialized, are you sure?')
|
63 |
|
64 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
65 |
model = model.half().to(device)
|
66 |
model.eval()
|
67 |
|
@@ -268,4 +274,4 @@ with block:
|
|
268 |
label='Text-to-3D Examples'
|
269 |
)
|
270 |
|
271 |
-
block.launch()
|
|
|
61 |
else:
|
62 |
print(f'[WARN] model randomly initialized, are you sure?')
|
63 |
|
64 |
+
if torch.backends.mps.is_available():
|
65 |
+
device = "mps"
|
66 |
+
elif torch.cuda.is_available():
|
67 |
+
device = "cuda"
|
68 |
+
else:
|
69 |
+
device = "cpu"
|
70 |
+
#device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
|
71 |
model = model.half().to(device)
|
72 |
model.eval()
|
73 |
|
|
|
274 |
label='Text-to-3D Examples'
|
275 |
)
|
276 |
|
277 |
+
block.launch()
|