biondizzle/vllm
1
0
Fork 0
Code Issues Pull Requests Actions 2 Packages Projects Releases Wiki Activity

[Bugfix] Missing thumbnail from NVLM-D processor (#14633)

Browse Source 
Signed-off-by: ameyanjarlekar <aanjarlekar@nvidia.com>
This commit is contained in:
ameyanjarlekar
2025-03-12 08:50:49 -07:00
committed by GitHub
parent c0c25e25fa
commit 4a754fcf15
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
vllm/model_executor/models/nvlm_d.py
View File

@@ -45,7 +45,7 @@ class NVLMProcessor(BaseInternVLProcessor):
raise NotImplementedError("Embedding inputs are not supported") raise NotImplementedError("Embedding inputs are not supported")
tile_pos_identifiers = [f"<tile_{i}>" for i in range(1, num_patches)] tile_pos_identifiers = [f"<tile_{i}>" for i in range(1, num_patches)]
if self.use_thumbnail and num_patches != 1: if self.use_thumbnail:
tile_pos_identifiers += ["<tile_global_thumbnail>"] tile_pos_identifiers += ["<tile_global_thumbnail>"]
context_size = feature_size // num_patches context_size = feature_size // num_patches