Update app.py
Browse files
app.py
CHANGED
|
@@ -82,8 +82,8 @@ from huggingface_hub import hf_hub_download
|
|
| 82 |
|
| 83 |
from utils.tool import NormalTransfer, get_background, get_render_cameras_video, load_mipmap, render_frames
|
| 84 |
|
| 85 |
-
device_0 = "cuda"
|
| 86 |
-
device_1 = "cuda"
|
| 87 |
resolution = 512
|
| 88 |
save_dir = "./outputs"
|
| 89 |
normal_transfer = NormalTransfer()
|
|
|
|
| 82 |
|
| 83 |
from utils.tool import NormalTransfer, get_background, get_render_cameras_video, load_mipmap, render_frames
|
| 84 |
|
| 85 |
+
device_0 = "cuda:0"
|
| 86 |
+
device_1 = "cuda:1"
|
| 87 |
resolution = 512
|
| 88 |
save_dir = "./outputs"
|
| 89 |
normal_transfer = NormalTransfer()
|