Yiheyihe commited on
Commit
41e9460
·
1 Parent(s): 17b2682

fix dinov2 loading

Browse files
Files changed (1) hide show
  1. src/model.py +1 -2
src/model.py CHANGED
@@ -42,8 +42,7 @@ class AffordanceModel:
42
  self.load_checkpoint(config.checkpoint_path)
43
 
44
  print("============ Building DINO model =============")
45
- torch_path = osp.join(osp.dirname(osp.dirname(__file__)), "data/torch_home")
46
- self.dinov2 = load_pretrained_dino(torch_path=torch_path)
47
 
48
  def load_checkpoint(self, checkpoint_path):
49
  """
 
42
  self.load_checkpoint(config.checkpoint_path)
43
 
44
  print("============ Building DINO model =============")
45
+ self.dinov2 = load_pretrained_dino()
 
46
 
47
  def load_checkpoint(self, checkpoint_path):
48
  """