Update modeling_minicpmv.py

modeling_minicpmv.py

@@ -302,11 +302,10 @@ class MiniCPMV(MiniCPMVPreTrainedModel):
             for pl in range(tensor.shape[1], max_x):
                 #print("here")
                 tensor = torch.cat((tensor, vector_reshaped), dim=1)
-                tensor = tensor.to(self.device)
-                batch[pl] = tensor
                 to_add.append(0)
+            tensor = tensor.to(self.device)
             print(tensor.shape, "UPDATED_SHAPE")
-            # batch[place] = tensor
+            batch[place] = tensor
             attention_mask.append(to_add)
         attention_mask = torch.tensor(attention_mask)
         print(attention_mask.shape)