Spaces:
Sleeping
Sleeping
HadiZayer commited on
Commit ·
8ff325d
1
Parent(s): 0f018d2
skip DINOv2 pretrained weights download (output is unused, checkpoint provides architecture weights)
Browse files
ldm/modules/encoders/modules.py
CHANGED
|
@@ -266,7 +266,7 @@ class DINOEmbedder(AbstractEncoder):
|
|
| 266 |
# embedding_size = embedding_sizes[dino_version]
|
| 267 |
letter = letter_map[dino_version]
|
| 268 |
# self.transformer = CLIPVisionModel.from_pretrained(version)
|
| 269 |
-
self.dino_model = torch.hub.load('facebookresearch/dinov2', f'dinov2_vit{letter}14_reg').cuda()
|
| 270 |
|
| 271 |
|
| 272 |
self.freeze()
|
|
|
|
| 266 |
# embedding_size = embedding_sizes[dino_version]
|
| 267 |
letter = letter_map[dino_version]
|
| 268 |
# self.transformer = CLIPVisionModel.from_pretrained(version)
|
| 269 |
+
self.dino_model = torch.hub.load('facebookresearch/dinov2', f'dinov2_vit{letter}14_reg', pretrained=False).cuda()
|
| 270 |
|
| 271 |
|
| 272 |
self.freeze()
|