Update app.py
Browse files
app.py
CHANGED
|
@@ -84,7 +84,7 @@ from huggingface_hub import hf_hub_download
|
|
| 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()
|
|
|
|
| 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:1"
|
| 88 |
resolution = 512
|
| 89 |
save_dir = "./outputs"
|
| 90 |
normal_transfer = NormalTransfer()
|