Fix rank for aspect_ratios

Signed-off-by: Benji Beck <[email protected]>

Author: bbeckca

vllm/model_executor/models/mllama4.py

@@ -778,6 +778,8 @@ def _parse_and_validate_image_input(
         flat_pixel_values = flatten_bn(pixel_values, concat=True)
         patches_per_image = flatten_bn(kwargs.pop("patches_per_image"))
         aspect_ratios = kwargs.pop("aspect_ratios")
+        if aspect_ratios.ndim == 3:
+            aspect_ratios = aspect_ratios.squeeze(1)
 
         return Llama4ImagePatchInputs(
             type="pixel_values",