Update modeling_moonvit.py

modeling_moonvit.py

@@ -180,7 +180,7 @@ class Learnable2DInterpPosEmb(nn.Module):
 
     def forward(self, x: torch.Tensor, grid_hws: torch.Tensor) -> torch.Tensor:
         pos_embs = []
-        for shape in grid_hws.tolist():
+        for shape in grid_hws[:, 1:].tolist():
             if shape == self.weight.shape[:-1]:
                 pos_embs.append(self.weight.flatten(end_dim=1))
             else: