update
Browse files
infer.py
CHANGED
|
@@ -160,7 +160,7 @@ def lotus_video(input_video, task_name, seed, device):
|
|
| 160 |
def lotus(image_input, task_name, seed, device):
|
| 161 |
if task_name == 'depth':
|
| 162 |
model_g = 'jingheya/lotus-depth-g-v1-0'
|
| 163 |
-
model_d = 'jingheya/lotus-depth-d-
|
| 164 |
else:
|
| 165 |
model_g = 'jingheya/lotus-normal-g-v1-0'
|
| 166 |
model_d = 'jingheya/lotus-normal-d-v1-0'
|
|
|
|
| 160 |
def lotus(image_input, task_name, seed, device):
|
| 161 |
if task_name == 'depth':
|
| 162 |
model_g = 'jingheya/lotus-depth-g-v1-0'
|
| 163 |
+
model_d = 'jingheya/lotus-depth-d-v1-1'
|
| 164 |
else:
|
| 165 |
model_g = 'jingheya/lotus-normal-g-v1-0'
|
| 166 |
model_d = 'jingheya/lotus-normal-d-v1-0'
|