Skip to content

Commit 8decbfa

Browse files
Test nemotron nano-v2 and nemotron nano-v3 separately, disable super-omni redundant tests (vllm-project#41616)
Signed-off-by: Netanel Haber <58652339+netanel-haber@users.noreply.github.com>
1 parent 62ba751 commit 8decbfa

1 file changed

Lines changed: 10 additions & 41 deletions

File tree

tests/models/registry.py

Lines changed: 10 additions & 41 deletions
Original file line numberDiff line numberDiff line change
@@ -1141,30 +1141,17 @@ def check_available_online(
11411141
"NemotronH_Nano_VL_V2": _HfExamplesInfo(
11421142
"nvidia/NVIDIA-Nemotron-Nano-12B-v2-VL-BF16",
11431143
max_model_len=4096,
1144-
# NemotronH layers are constructed via `hybrid_override_pattern`:
1144+
# NemotronH layers are constructed via `hybrid_override_pattern`
11451145
use_original_num_layers=True,
11461146
hf_overrides={
1147-
"vision_config": PretrainedConfig(
1148-
args={
1149-
"min_num_patches": 1, # Trigger image dynamic res
1150-
"max_num_patches": 12,
1151-
"model": "vit_huge_patch16_224",
1152-
},
1153-
# Trigger conv3d:
1154-
video_temporal_patch_size=2,
1155-
),
1156-
"text_config": {
1157-
"num_hidden_layers": 2,
1158-
"hybrid_override_pattern": "M*",
1159-
},
1147+
"text_config": {"num_hidden_layers": 2, "hybrid_override_pattern": "M*"},
11601148
},
11611149
trust_remote_code=True,
11621150
),
1163-
# NemotronH_Nano_Omni_Reasoning_V3 is an alias for NemotronH_Nano_VL_V2
1164-
# Use the same registry test as NemotronH_Nano_VL_V2 above
11651151
"NemotronH_Nano_Omni_Reasoning_V3": _HfExamplesInfo(
1166-
"nvidia/NVIDIA-Nemotron-Nano-12B-v2-VL-BF16",
1152+
"nvidia/Nemotron-3-Nano-Omni-30B-A3B-Reasoning-BF16",
11671153
max_model_len=4096,
1154+
# NemotronH layers are constructed via `hybrid_override_pattern`
11681155
use_original_num_layers=True,
11691156
hf_overrides={
11701157
"vision_config": PretrainedConfig(
@@ -1174,35 +1161,17 @@ def check_available_online(
11741161
"model": "vit_huge_patch16_224",
11751162
},
11761163
video_temporal_patch_size=2,
1164+
# TODO(nhaber): This is `true` in the official `config.json`,
1165+
# but this causes a processor exception in the tests due to a known bug
1166+
# with mixed-resolution video when `true`. To be resolved.
1167+
video_maintain_aspect_ratio=False,
11771168
),
1178-
"text_config": {
1179-
"num_hidden_layers": 2,
1180-
"hybrid_override_pattern": "M*",
1181-
},
1169+
"text_config": {"num_hidden_layers": 2, "hybrid_override_pattern": "M*"},
11821170
},
11831171
trust_remote_code=True,
11841172
),
1185-
# NemotronH_Super_Omni_Reasoning_V3 is an alias for NemotronH_Nano_VL_V2 as well
1186-
# Use the same registry test as NemotronH_Nano_VL_V2 above
11871173
"NemotronH_Super_Omni_Reasoning_V3": _HfExamplesInfo(
1188-
"nvidia/NVIDIA-Nemotron-Nano-12B-v2-VL-BF16",
1189-
max_model_len=4096,
1190-
use_original_num_layers=True,
1191-
hf_overrides={
1192-
"vision_config": PretrainedConfig(
1193-
args={
1194-
"min_num_patches": 1,
1195-
"max_num_patches": 12,
1196-
"model": "vit_huge_patch16_224",
1197-
},
1198-
video_temporal_patch_size=2,
1199-
),
1200-
"text_config": {
1201-
"num_hidden_layers": 2,
1202-
"hybrid_override_pattern": "M*",
1203-
},
1204-
},
1205-
trust_remote_code=True,
1174+
"nvidia/Nemotron-3-Nano-Omni-30B-A3B-Reasoning-BF16", is_available_online=False
12061175
),
12071176
"OpenCUAForConditionalGeneration": _HfExamplesInfo(
12081177
"xlangai/OpenCUA-7B",

0 commit comments

Comments
 (0)