Spaces:

haodongli
/

Lotus_Depth

Running on Zero

haodongli commited on Jan 17

Commit

b5ad4c4

1 Parent(s): f562867

update lotus-depth-g-v2-1-disparity

Files changed (1) hide show

infer.py CHANGED Viewed

@@ -116,7 +116,7 @@ def infer_pipe_video(pipe, test_image, task_name, generator, device, latents=Non
 def load_pipe(task_name, device):
     if task_name == 'depth':
-        model_g = 'jingheya/lotus-depth-g-v2-0-disparity'
         model_d = 'jingheya/lotus-depth-d-v2-0-disparity'
     else:
         model_g = 'jingheya/lotus-normal-g-v1-0'

 def load_pipe(task_name, device):
     if task_name == 'depth':
+        model_g = 'jingheya/lotus-depth-g-v2-1-disparity'
         model_d = 'jingheya/lotus-depth-d-v2-0-disparity'
     else:
         model_g = 'jingheya/lotus-normal-g-v1-0'