@@ -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