Update Model/dino/dino.py

Model/dino/dino.py

@@ -15,7 +15,7 @@ class DinoV2(nn.Module):
     def __init__(self, model_name):
         super().__init__()
         self.vision_encoder = torch.hub.load('facebookresearch/dinov2', model_name)
-        self.vision_encoder = self.vision_encoder.eval().cuda().half()
+        self.vision_encoder = self.vision_encoder.eval()
         return
 
     def forward(self, x):
@@ -24,6 +24,6 @@ class DinoV2(nn.Module):
     def get_output_dim(self):
         with torch.no_grad():
             dummpy_input_image = preprocess(Image.fromarray(numpy.zeros((512, 512, 3), dtype=numpy.uint8))).to(
-                next(self.parameters()).device).half()
+                next(self.parameters()).device)
             encoder_output_size = self.vision_encoder(dummpy_input_image.unsqueeze(0)).shape[-1]
         return encoder_output_size