Skip to content

Commit 883934f

Browse files
kirit93kirit93
authored
Merge pull request #401 from mikeknep/ndd-25-12/mknepper
Data designer updates for 25.12
2 parents fe694ba + eae7466 commit 883934f

File tree

6 files changed

+48
-24
lines changed

6 files changed

+48
-24
lines changed

‎nemo/NeMo-Data-Designer/managed-service-tutorials/getting-started/1-the-basics.ipynb‎

Lines changed: 3 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -103,9 +103,7 @@
103103
"metadata": {},
104104
"source": [
105105
"**Note**:\n",
106-
"The NeMo Data Designer Managed service has models available for you to use as well. You can use these models by referencing the appropriate model_alias for them.\n",
107-
"\n",
108-
"Please visit https://build.nvidia.com/nemo/data-designer to see the full list of models and their model aliases.\n"
106+
"The NeMo Data Designer Managed service has access to specific models. Please visit https://build.nvidia.com/nemo/data-designer to see the latest list of which models are available.\n"
109107
]
110108
},
111109
{
@@ -114,8 +112,8 @@
114112
"metadata": {},
115113
"outputs": [],
116114
"source": [
117-
"model_alias = \"nemotron-nano-v2\"\n",
118-
"model_id = \"nvidia/nvidia-nemotron-nano-9b-v2\"\n"
115+
"model_alias = \"nemotron-3-nano\"\n",
116+
"model_id = \"nvidia/nemotron-3-nano-30b-a3b\"\n"
119117
]
120118
},
121119
{

‎nemo/NeMo-Data-Designer/managed-service-tutorials/getting-started/2-structured-outputs-and-jinja-expressions.ipynb‎

Lines changed: 19 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -36,7 +36,9 @@
3636
" CategorySamplerParams,\n",
3737
" DataDesignerConfigBuilder,\n",
3838
" ExpressionColumnConfig,\n",
39+
" InferenceParameters,\n",
3940
" LLMStructuredColumnConfig,\n",
41+
" ModelConfig,\n",
4042
" NeMoDataDesignerClient,\n",
4143
" PersonSamplerParams,\n",
4244
" SamplerColumnConfig,\n",
@@ -107,9 +109,7 @@
107109
"metadata": {},
108110
"source": [
109111
"**Note**:\n",
110-
"The NeMo Data Designer Managed service has models available for you to use as well. You can use these models by referencing the appropriate model_alias for them.\n",
111-
"\n",
112-
"Please visit https://build.nvidia.com/nemo/data-designer to see the full list of models and their model aliases.\n"
112+
"The NeMo Data Designer Managed service has access to specific models. Please visit https://build.nvidia.com/nemo/data-designer to see the latest list of which models are available."
113113
]
114114
},
115115
{
@@ -118,7 +118,8 @@
118118
"metadata": {},
119119
"outputs": [],
120120
"source": [
121-
"model_alias = \"nemotron-nano-v2\"\n"
121+
"model_alias = \"nemotron-3-nano\"\n",
122+
"model_id = \"nvidia/nemotron-3-nano-30b-a3b\""
122123
]
123124
},
124125
{
@@ -127,7 +128,20 @@
127128
"metadata": {},
128129
"outputs": [],
129130
"source": [
130-
"config_builder = DataDesignerConfigBuilder()\n"
131+
"config_builder = DataDesignerConfigBuilder(\n",
132+
" model_configs=[\n",
133+
" ModelConfig(\n",
134+
" alias=model_alias,\n",
135+
" provider=\"nvidiabuild\",\n",
136+
" model=model_id,\n",
137+
" inference_parameters=InferenceParameters(\n",
138+
" max_tokens=1024,\n",
139+
" temperature=0.6,\n",
140+
" top_p=0.95,\n",
141+
" ),\n",
142+
" ),\n",
143+
" ]\n",
144+
")"
131145
]
132146
},
133147
{

‎nemo/NeMo-Data-Designer/managed-service-tutorials/getting-started/3-seeding-with-a-dataset.ipynb‎

Lines changed: 19 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -34,6 +34,8 @@
3434
"\n",
3535
"from nemo_microservices.data_designer.essentials import (\n",
3636
" DataDesignerConfigBuilder,\n",
37+
" InferenceParameters,\n",
38+
" ModelConfig,\n",
3739
" NeMoDataDesignerClient,\n",
3840
" SeedDatasetReference,\n",
3941
")\n"
@@ -101,9 +103,7 @@
101103
"metadata": {},
102104
"source": [
103105
"**Note**:\n",
104-
"The NeMo Data Designer Managed service has models available for you to use as well. You can use these models by referencing the appropriate model_alias for them.\n",
105-
"\n",
106-
"Please visit https://build.nvidia.com/nemo/data-designer to see the full list of models and their model aliases.\n"
106+
"The NeMo Data Designer Managed service has access to specific models. Please visit https://build.nvidia.com/nemo/data-designer to see the latest list of which models are available.\n"
107107
]
108108
},
109109
{
@@ -112,7 +112,8 @@
112112
"metadata": {},
113113
"outputs": [],
114114
"source": [
115-
"model_alias = \"nemotron-nano-v2\""
115+
"model_alias = \"nemotron-3-nano\"\n",
116+
"model_id = \"nvidia/nemotron-3-nano-30b-a3b\""
116117
]
117118
},
118119
{
@@ -121,7 +122,20 @@
121122
"metadata": {},
122123
"outputs": [],
123124
"source": [
124-
"config_builder = DataDesignerConfigBuilder()"
125+
"config_builder = DataDesignerConfigBuilder(\n",
126+
" model_configs=[\n",
127+
" ModelConfig(\n",
128+
" alias=model_alias,\n",
129+
" provider=\"nvidiabuild\",\n",
130+
" model=model_id,\n",
131+
" inference_parameters=InferenceParameters(\n",
132+
" max_tokens=1024,\n",
133+
" temperature=0.6,\n",
134+
" top_p=0.95,\n",
135+
" ),\n",
136+
" ),\n",
137+
" ]\n",
138+
")"
125139
]
126140
},
127141
{

‎nemo/NeMo-Data-Designer/self-hosted-tutorials/getting-started/1-the-basics.ipynb‎

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -99,10 +99,10 @@
9999
"MODEL_PROVIDER = \"nvidiabuild\"\n",
100100
"\n",
101101
"# The model ID is from build.nvidia.com.\n",
102-
"MODEL_ID = \"nvidia/nvidia-nemotron-nano-9b-v2\"\n",
102+
"MODEL_ID = \"nvidia/nemotron-3-nano-30b-a3b\"\n",
103103
"\n",
104104
"# We choose this alias to be descriptive for our use case.\n",
105-
"MODEL_ALIAS = \"nemotron-nano-v2\"\n",
105+
"MODEL_ALIAS = \"nemotron-nano-v3\"\n",
106106
"\n",
107107
"# This sets reasoning to False for the nemotron-nano-v2 model.\n",
108108
"SYSTEM_PROMPT = \"/no_think\"\n",
@@ -164,7 +164,7 @@
164164
"metadata": {},
165165
"outputs": [],
166166
"source": [
167-
"config_builder.info.sampler_table"
167+
"config_builder.info.display(\"samplers\")"
168168
]
169169
},
170170
{

‎nemo/NeMo-Data-Designer/self-hosted-tutorials/getting-started/2-structured-outputs-and-jinja-expressions.ipynb‎

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -101,10 +101,10 @@
101101
"MODEL_PROVIDER = \"nvidiabuild\"\n",
102102
"\n",
103103
"# The model ID is from build.nvidia.com.\n",
104-
"MODEL_ID = \"nvidia/nvidia-nemotron-nano-9b-v2\"\n",
104+
"MODEL_ID = \"nvidia/nemotron-3-nano-30b-a3b\"\n",
105105
"\n",
106106
"# We choose this alias to be descriptive for our use case.\n",
107-
"MODEL_ALIAS = \"nemotron-nano-v2\"\n",
107+
"MODEL_ALIAS = \"nemotron-nano-v3\"\n",
108108
"\n",
109109
"# This sets reasoning to False for the nemotron-nano-v2 model.\n",
110110
"SYSTEM_PROMPT = \"/no_think\"\n",

‎nemo/NeMo-Data-Designer/self-hosted-tutorials/getting-started/3-seeding-with-a-dataset.ipynb‎

Lines changed: 2 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -94,10 +94,10 @@
9494
"MODEL_PROVIDER = \"nvidiabuild\"\n",
9595
"\n",
9696
"# The model ID is from build.nvidia.com.\n",
97-
"MODEL_ID = \"nvidia/nvidia-nemotron-nano-9b-v2\"\n",
97+
"MODEL_ID = \"nvidia/nemotron-3-nano-30b-a3b\"\n",
9898
"\n",
9999
"# We choose this alias to be descriptive for our use case.\n",
100-
"MODEL_ALIAS = \"nemotron-nano-v2\"\n",
100+
"MODEL_ALIAS = \"nemotron-nano-v3\"\n",
101101
"\n",
102102
"# This sets reasoning to False for the nemotron-nano-v2 model.\n",
103103
"SYSTEM_PROMPT = \"/no_think\"\n",
@@ -237,14 +237,12 @@
237237
" name=\"patient_sampler\",\n",
238238
" column_type=\"sampler\",\n",
239239
" sampler_type=\"person\",\n",
240-
" drop=True,\n",
241240
")\n",
242241
"\n",
243242
"config_builder.add_column(\n",
244243
" name=\"doctor_sampler\",\n",
245244
" column_type=\"sampler\",\n",
246245
" sampler_type=\"person\",\n",
247-
" drop=True,\n",
248246
")\n",
249247
"\n",
250248
"config_builder.add_column(\n",

0 commit comments

Comments
 (0)