haodongli commited on
Commit
10a5999
·
1 Parent(s): 4d23710

update lotus-normal-g-v1-1 and lotus-normal-d-v1-1

Browse files
Files changed (1) hide show
  1. infer.py +2 -2
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-0'
82
- model_d = 'jingheya/lotus-normal-d-v1-0'
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(