yamildiego commited on
Commit
101eab1
·
1 Parent(s): 5366461

load model externally

Browse files
Files changed (3) hide show
  1. .DS_Store +0 -0
  2. checkpoints/.DS_Store +0 -0
  3. depth_anything/dpt.py +1 -1
.DS_Store CHANGED
Binary files a/.DS_Store and b/.DS_Store differ
 
checkpoints/.DS_Store CHANGED
Binary files a/checkpoints/.DS_Store and b/checkpoints/.DS_Store differ
 
depth_anything/dpt.py CHANGED
@@ -137,7 +137,7 @@ class DPTHead(nn.Module):
137
 
138
 
139
  class DPT_DINOv2(nn.Module):
140
- def __init__(self, encoder='vitl', features=256, out_channels=[256, 512, 1024, 1024], use_bn=False, use_clstoken=False, localhub=True):
141
  super(DPT_DINOv2, self).__init__()
142
 
143
  assert encoder in ['vits', 'vitb', 'vitl']
 
137
 
138
 
139
  class DPT_DINOv2(nn.Module):
140
+ def __init__(self, encoder='vitl', features=256, out_channels=[256, 512, 1024, 1024], use_bn=False, use_clstoken=False, localhub=False):
141
  super(DPT_DINOv2, self).__init__()
142
 
143
  assert encoder in ['vits', 'vitb', 'vitl']