Update modeling_GOT.py

modeling_GOT.py

@@ -237,7 +237,6 @@ class GOTQwenModel(Qwen2Model):
             
             for image in images:
                 P, C, H, W = image.shape
-                print(image.shape)
                 if P == 1:
                     with torch.set_grad_enabled(False):
                         cnn_feature = vision_tower_high(image)
@@ -250,6 +249,7 @@ class GOTQwenModel(Qwen2Model):
                     image_patches_features = []
                     for image_patch in image_patches:
                         image_p = torch.stack([image_patch])
+                        print(image_p.shape)
                         with torch.set_grad_enabled(False):
                             cnn_feature_p = vision_tower_high(image_p)
                             cnn_feature_p = cnn_feature_p.flatten(2).permute(0, 2, 1)