Unverified Commit 4a754fcf authored by ameyanjarlekar's avatar ameyanjarlekar Committed by GitHub
Browse files

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


Signed-off-by: default avatarameyanjarlekar <aanjarlekar@nvidia.com>
parent c0c25e25
......@@ -45,7 +45,7 @@ class NVLMProcessor(BaseInternVLProcessor):
raise NotImplementedError("Embedding inputs are not supported")
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>"]
context_size = feature_size // num_patches
......
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