Spaces:

Jiaqi-hkust
/

hawk

Runtime error

Jiaqi-hkust commited on Feb 24

Commit

6fb09d0

verified ·

1 Parent(s): 7aeb916

Update hawk/models/eva_vit.py

Files changed (1) hide show

hawk/models/eva_vit.py CHANGED Viewed

@@ -347,6 +347,7 @@ class VisionTransformer(nn.Module):
 #             return x[:, 0]
     def forward(self, x):
         x = self.forward_features(x)
 #         x = self.head(x)
         return x

 #             return x[:, 0]
     def forward(self, x):
+        x = x.float()
         x = self.forward_features(x)
 #         x = self.head(x)
         return x