Hugging Face's logo

Spaces:
NightRaven109
/
PBRFusion4DepthDemo-InstNorm
Running on Zero

App Files Community
NightRaven109 commited on
Commit
ad041dc
·
verified ·
1 Parent(s): 54c84c8

Update infer.py

Browse files
Files changed (1) hide show
  1. infer.py +1 -1
infer.py CHANGED
@@ -118,7 +118,7 @@ def infer_pipe_video(pipe, test_image, task_name, generator, device, latents=Non
118
  def load_pipe(task_name, device):
119
  if task_name == 'depth':
120
  model_g = 'jingheya/lotus-depth-g-v2-1-disparity'
121
- model_d = 'NightRaven109/LotusD'
122
  else:
123
  model_g = 'jingheya/lotus-normal-g-v1-0'
124
  model_d = 'jingheya/lotus-normal-d-v1-0'
 
118
  def load_pipe(task_name, device):
119
  if task_name == 'depth':
120
  model_g = 'jingheya/lotus-depth-g-v2-1-disparity'
121
+ model_d = 'NightRaven109/LotusDInstNorm'
122
  else:
123
  model_g = 'jingheya/lotus-normal-g-v1-0'
124
  model_d = 'jingheya/lotus-normal-d-v1-0'