Skip to content

Commit 87b836b

Browse files
authored
Bugfix for PixtralHF models without spatial_merge_size (#16513)
Signed-off-by: mgoin <mgoin64@gmail.com>
1 parent 56c76c2 commit 87b836b

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

vllm/model_executor/models/pixtral.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -926,8 +926,9 @@ def get_image_size(self) -> int:
926926
return self.vision_config.image_size
927927

928928
def get_patch_size(self) -> int:
929-
return (self.vision_config.patch_size *
930-
self.vision_config.spatial_merge_size)
929+
spatial_merge_size = getattr(self.vision_config, "spatial_merge_size",
930+
1)
931+
return (self.vision_config.patch_size * spatial_merge_size)
931932

932933
def get_patch_grid_length(self) -> int:
933934
image_size, patch_size = self.get_image_size(), self.get_patch_size()

0 commit comments

Comments
 (0)