Skip to content

GitLab

Unverified Commit 87b836ba authored by Michael Goin's avatar Michael Goin Committed by GitHub
Browse files

Bugfix for PixtralHF models without spatial_merge_size (#16513) 


Signed-off-by: default avatarmgoin <mgoin64@gmail.com>
parent 56c76c2e
Show whitespace changes
Inline Side-by-side
Showing with 3 additions and 2 deletions
vllm/model_executor/models/pixtral.py
View file @ 87b836ba
......@@ -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()
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment