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