LTT commited on
Commit
79847fa
·
verified ·
1 Parent(s): a468b1f

Update app.py

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