Spaces:
Running
on
Zero
Running
on
Zero
Stanislaw Szymanowicz
commited on
Commit
·
0b4669a
1
Parent(s):
3f12c93
Do not set device
Browse files
app.py
CHANGED
|
@@ -33,7 +33,7 @@ def main():
|
|
| 33 |
device = "cuda:0"
|
| 34 |
else:
|
| 35 |
device = "cpu"
|
| 36 |
-
torch.cuda.set_device(device)
|
| 37 |
|
| 38 |
model_cfg = OmegaConf.load(
|
| 39 |
os.path.join(
|
|
|
|
| 33 |
device = "cuda:0"
|
| 34 |
else:
|
| 35 |
device = "cpu"
|
| 36 |
+
# torch.cuda.set_device(device)
|
| 37 |
|
| 38 |
model_cfg = OmegaConf.load(
|
| 39 |
os.path.join(
|