Fix number of image tokens

image_processing_phi3_v.py

@@ -250,7 +250,7 @@ class Phi3VImageProcessor(BaseImageProcessor):
 
         # [(3, h, w)], where h, w is multiple of 336
         shapes = [[im.size(1), im.size(2)] for im in hd_images]
-        num_img_tokens = [int((h//336*w//336+1)*144 + 1 + (h//336+1)*12) for h, w in shapes]
+        num_img_tokens = [int(((h//336)*(w//336)+1)*144 + 1 + (h//336+1)*12) for h, w in shapes]
         # reshape to channel dimension -> (num_images, num_crops, 3, 336, 336)
         # (1, 3, h//336, 336, w//336, 336) -> (1, h//336, w//336, 3, 336, 336) -> (h//336*w//336, 3, 336, 336)
         hd_images_reshape = [im.reshape(1, 3, h//336, 336, w//336, 336).permute(0,2,4,1,3,5).reshape(-1, 3, 336, 336).contiguous() for im, (h, w) in zip(hd_images, shapes)]