[Bugfix] Standardize getting number of image patches/tokens (#34358)

Signed-off-by: DarkLight1337 <tlleungac@connect.ust.hk>
2026-02-13 12:47:01 +08:00
parent 6afa587d31
commit 372b2e762a
29 changed files with 319 additions and 331 deletions
--- a/vllm/model_executor/models/molmo.py
+++ b/vllm/model_executor/models/molmo.py
@@ -1224,11 +1224,8 @@ class MolmoProcessingInfo(BaseProcessingInfo):
        *,
        image_width: int,
        image_height: int,
-        processor: MolmoProcessorWrapper | None,
+        processor: MolmoProcessorWrapper,
    ) -> int:
-        if processor is None:
-            processor = self.get_hf_processor()
-
        ncols, nrows = processor.get_patches_grid_size(
            image_width=image_width,
            image_height=image_height,