Spaces:
Runtime error
Runtime error
update lotus-normal-g-v1-1 and lotus-normal-d-v1-1
Browse files
infer.py
CHANGED
|
@@ -78,8 +78,8 @@ def lotus_video(input_video, task_name, seed, device):
|
|
| 78 |
model_g = 'jingheya/lotus-depth-g-v1-0'
|
| 79 |
model_d = 'jingheya/lotus-depth-d-v1-1'
|
| 80 |
else:
|
| 81 |
-
model_g = 'jingheya/lotus-normal-g-v1-
|
| 82 |
-
model_d = 'jingheya/lotus-normal-d-v1-
|
| 83 |
|
| 84 |
dtype = torch.float16
|
| 85 |
pipe_g = LotusGPipeline.from_pretrained(
|
|
|
|
| 78 |
model_g = 'jingheya/lotus-depth-g-v1-0'
|
| 79 |
model_d = 'jingheya/lotus-depth-d-v1-1'
|
| 80 |
else:
|
| 81 |
+
model_g = 'jingheya/lotus-normal-g-v1-1'
|
| 82 |
+
model_d = 'jingheya/lotus-normal-d-v1-1'
|
| 83 |
|
| 84 |
dtype = torch.float16
|
| 85 |
pipe_g = LotusGPipeline.from_pretrained(
|