HadiZayer commited on
Commit
8ff325d
·
1 Parent(s): 0f018d2

skip DINOv2 pretrained weights download (output is unused, checkpoint provides architecture weights)

Browse files
Files changed (1) hide show
  1. ldm/modules/encoders/modules.py +1 -1
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()