Update visual.py

visual.py

@@ -130,7 +130,7 @@ class EVA2CLIPModel(nn.Module):
         x = self.transformer(x)
         x = x[:, 1:]
         x = self.linear_proj(x + self.pos_embed.to(x.device).unsqueeze(0))
-        boi = self.boi.expand(x.shape[0], -1, -1)
-        eoi = self.eoi.expand(x.shape[0], -1, -1)
+        boi = self.boi.to(x.device).expand(x.shape[0], -1, -1)
+        eoi = self.eoi.to(x.device).expand(x.shape[0], -1, -1)
         x = torch.cat((boi, x, eoi), dim=1)
         return x