Spaces:
Sleeping
Sleeping
fix dinov2 loading
Browse files- 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 |
-
|
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 |
"""
|