From 87b836ba777c07acf35da802be0aacbfc51da3b5 Mon Sep 17 00:00:00 2001 From: Michael Goin Date: Fri, 11 Apr 2025 17:32:22 -0600 Subject: [PATCH] Bugfix for PixtralHF models without spatial_merge_size (#16513) Signed-off-by: mgoin --- vllm/model_executor/models/pixtral.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/vllm/model_executor/models/pixtral.py b/vllm/model_executor/models/pixtral.py index ee1e7713..38e140a9 100644 --- a/vllm/model_executor/models/pixtral.py +++ b/vllm/model_executor/models/pixtral.py @@ -926,8 +926,9 @@ class PixtralHFEncoderInfo(VisionEncoderInfo[PixtralVisionConfig]): return self.vision_config.image_size def get_patch_size(self) -> int: - return (self.vision_config.patch_size * - self.vision_config.spatial_merge_size) + spatial_merge_size = getattr(self.vision_config, "spatial_merge_size", + 1) + return (self.vision_config.patch_size * spatial_merge_size) def get_patch_grid_length(self) -> int: image_size, patch_size = self.get_image_size(), self.get_patch_size()