xinyun/vllm
1
0
Fork 0
mirror of https://git.datalinker.icu/vllm-project/vllm.git synced 2025-12-10 16:45:43 +08:00
Code Issues Packages Projects Releases Wiki Activity

[Model][Gemma3] Simplify image input validation (#18710)

Browse Source 
Signed-off-by: Lukas Geiger <lukas.geiger94@gmail.com>
This commit is contained in:
Lukas Geiger 2025-05-27 04:13:37 +01:00 committed by GitHub
parent 27bebcd897
commit 0eebd74842
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 6 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
vllm/model_executor/models/gemma3_mm.py
View File

@ -504,18 +504,12 @@ class Gemma3ForConditionalGeneration(nn.Module, SupportsMultiModal, SupportsPP,
return next(self.parameters()).dtype return next(self.parameters()).dtype
def _validate_pixel_values(self, data: torch.Tensor) -> torch.Tensor: def _validate_pixel_values(self, data: torch.Tensor) -> torch.Tensor:
h = w = self.config.vision_config.image_size image_size = self.config.vision_config.image_size
expected_dims = (3, h, w) expected_dims = (3, image_size, image_size)
if data.shape[1:] != expected_dims:
def _validate_shape(d: torch.Tensor): raise ValueError(
if d.shape != expected_dims: "The expected shape of pixel values per image per batch is "
raise ValueError( f"{expected_dims}. You supplied {tuple(data.shape)}.")
"The expected shape of pixel values per image per batch "
f"is {expected_dims}. You supplied {tuple(d.shape)}.")
for d in data:
_validate_shape(d)
return data return data
def _parse_and_validate_image_input( def _parse_and_validate_image_input(