Spaces:
Runtime error
Runtime error
Commit ·
26b43e9
1
Parent(s): 0241451
change GPU
Browse files
app.py
CHANGED
|
@@ -1,6 +1,6 @@
|
|
| 1 |
import os
|
| 2 |
|
| 3 |
-
os.environ["CUDA_VISIBLE_DEVICES"] = "
|
| 4 |
import copy
|
| 5 |
import torch
|
| 6 |
from torchvision.io import write_video
|
|
|
|
| 1 |
import os
|
| 2 |
|
| 3 |
+
os.environ["CUDA_VISIBLE_DEVICES"] = "0"
|
| 4 |
import copy
|
| 5 |
import torch
|
| 6 |
from torchvision.io import write_video
|