Skip to content

Commit 93ae875

Browse files
authored
feat: update default nvidia-vision model to nemotron-3-nano-omni (#583)
* update nvidia-vision * update temp
1 parent 482ab5a commit 93ae875

4 files changed

Lines changed: 9 additions & 5 deletions

File tree

docs/concepts/models/default-model-settings.md

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -45,7 +45,7 @@ The following model configurations are automatically available when `NVIDIA_API_
4545
|-------|-------|----------|---------------------|
4646
| `nvidia-text` | `nvidia/nemotron-3-nano-30b-a3b` | General text generation | `temperature=1.0, top_p=1.0` |
4747
| `nvidia-reasoning` | `nvidia/nemotron-3-super-120b-a12b` | Reasoning and analysis tasks | `temperature=1.0, top_p=0.95, extra_body={"reasoning_effort": "medium"}` |
48-
| `nvidia-vision` | `nvidia/nemotron-nano-12b-v2-vl` | Vision and image understanding | `temperature=0.85, top_p=0.95` |
48+
| `nvidia-vision` | `nvidia/nemotron-3-nano-omni-30b-a3b-reasoning` | Vision and image understanding | `temperature=0.60, top_p=0.95` |
4949
| `nvidia-embedding` | `nvidia/llama-3.2-nv-embedqa-1b-v2` | Text embeddings | `encoding_format="float", extra_body={"input_type": "query"}` |
5050

5151

docs/concepts/models/model-configs.md

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -81,10 +81,10 @@ model_configs = [
8181
# Vision tasks
8282
dd.ModelConfig(
8383
alias="vision-model",
84-
model="nvidia/nemotron-nano-12b-v2-vl",
84+
model="nvidia/nemotron-3-nano-omni-30b-a3b-reasoning",
8585
provider="nvidia",
8686
inference_parameters=dd.ChatCompletionInferenceParams(
87-
temperature=0.7,
87+
temperature=0.60,
8888
top_p=0.95,
8989
max_tokens=2048,
9090
),

packages/data-designer-config/src/data_designer/config/utils/constants.py

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -341,6 +341,7 @@ class NordColor(Enum):
341341
"top_p": 0.95,
342342
"extra_body": {"reasoning_effort": "medium"},
343343
}
344+
NEMOTRON_3_NANO_OMNI_30B_A3B_REASONING_INFERENCE_PARAMS = {"temperature": 0.60, "top_p": 0.95}
344345
GPT5_INFERENCE_PARAMS = {"extra_body": {"reasoning_effort": "medium"}}
345346

346347
PREDEFINED_PROVIDERS_MODEL_MAP = {
@@ -353,7 +354,10 @@ class NordColor(Enum):
353354
"model": "nvidia/nemotron-3-super-120b-a12b",
354355
"inference_parameters": NEMOTRON_3_SUPER_120B_A12B_INFERENCE_PARAMS,
355356
},
356-
"vision": {"model": "nvidia/nemotron-nano-12b-v2-vl", "inference_parameters": DEFAULT_VISION_INFERENCE_PARAMS},
357+
"vision": {
358+
"model": "nvidia/nemotron-3-nano-omni-30b-a3b-reasoning",
359+
"inference_parameters": NEMOTRON_3_NANO_OMNI_30B_A3B_REASONING_INFERENCE_PARAMS,
360+
},
357361
"embedding": {
358362
"model": "nvidia/llama-3.2-nv-embedqa-1b-v2",
359363
"inference_parameters": DEFAULT_EMBEDDING_INFERENCE_PARAMS | {"extra_body": {"input_type": "query"}},

packages/data-designer-config/tests/config/test_default_model_settings.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -63,7 +63,7 @@ def test_get_builtin_model_configs():
6363
assert builtin_model_configs[1].model == "nvidia/nemotron-3-super-120b-a12b"
6464
assert builtin_model_configs[1].provider == "nvidia"
6565
assert builtin_model_configs[2].alias == "nvidia-vision"
66-
assert builtin_model_configs[2].model == "nvidia/nemotron-nano-12b-v2-vl"
66+
assert builtin_model_configs[2].model == "nvidia/nemotron-3-nano-omni-30b-a3b-reasoning"
6767
assert builtin_model_configs[2].provider == "nvidia"
6868
assert builtin_model_configs[3].alias == "nvidia-embedding"
6969
assert builtin_model_configs[3].model == "nvidia/llama-3.2-nv-embedqa-1b-v2"

0 commit comments

Comments
 (0)