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

[Bugfix] Fix Nemotron VL image processing (#22739)

Browse Source 
Co-authored-by: ducviet00-h2 <viet.d.hoang@h2corporation.jp>
This commit is contained in:
Duc-Viet Hoang
2025-08-13 17:11:36 +07:00
committed by GitHub
parent 9e7e5baaa8
commit a01e0018b5
2 changed files with 190 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
tests/models/multimodal/processing/test_nemotron_vl.py
View File

@@ -23,15 +23,15 @@ def _get_expected_num_patches(
min_num: int,
max_num: int,
):
from vllm.model_executor.models.internvl import (
calculate_internvl_targets, get_internvl_target_ratios)
from vllm.model_executor.models.nemotron_vl import (
calculate_nemotron_vl_targets, get_nemotron_vl_target_ratios)
width, height = image.size
blocks, _, _ = calculate_internvl_targets(
blocks, _, _ = calculate_nemotron_vl_targets(
orig_width=width,
orig_height=height,
target_ratios=get_internvl_target_ratios(
target_ratios=get_nemotron_vl_target_ratios(
min_num,
max_num,
),