SilentWraith commited on
Commit
1b3fa55
·
verified ·
1 Parent(s): 89e0320

Update src/condition.py

Browse files
Files changed (1) hide show
  1. src/condition.py +1 -1
src/condition.py CHANGED
@@ -43,7 +43,7 @@ class Condition(object):
43
 
44
  depth_pipe = pipeline(
45
  task="depth-estimation",
46
- model="LiheYoung/depth-anything-small-hf",
47
  device="cuda",
48
  )
49
  source_image = raw_img.convert("RGB")
 
43
 
44
  depth_pipe = pipeline(
45
  task="depth-estimation",
46
+ model="LiheYoung/depth-anything-large-hf",
47
  device="cuda",
48
  )
49
  source_image = raw_img.convert("RGB")