yamildiego commited on
Commit
ba2fff5
·
1 Parent(s): b55d430

fix path tourch

Browse files
Files changed (1) hide show
  1. depth_anything/dpt.py +2 -2
depth_anything/dpt.py CHANGED
@@ -144,9 +144,9 @@ class DPT_DINOv2(nn.Module):
144
 
145
  # in case the Internet connection is not stable, please load the DINOv2 locally
146
  if localhub:
147
- self.pretrained = torch.hub.load('torchhub/facebookresearch_dinov2_main', 'dinov2_{:}14'.format(encoder), source='local', pretrained=False)
148
  else:
149
- self.pretrained = torch.hub.load('facebookresearch/dinov2', 'dinov2_{:}14'.format(encoder))
150
 
151
  dim = self.pretrained.blocks[0].attn.qkv.in_features
152
 
 
144
 
145
  # in case the Internet connection is not stable, please load the DINOv2 locally
146
  if localhub:
147
+ self.pretrained = torch.hub.load('../torchhub/facebookresearch_dinov2_main', 'dinov2_{:}14'.format(encoder), source='local', pretrained=False)
148
  else:
149
+ self.pretrained = torch.hub.load('../facebookresearch/dinov2', 'dinov2_{:}14'.format(encoder))
150
 
151
  dim = self.pretrained.blocks[0].attn.qkv.in_features
152