diff --git a/docs/notebooks/nvidia/Llama_Stack_NVIDIA_E2E_Flow.ipynb b/docs/notebooks/nvidia/Llama_Stack_NVIDIA_E2E_Flow.ipynb
index 134f572fd..fbf78018a 100644
--- a/docs/notebooks/nvidia/Llama_Stack_NVIDIA_E2E_Flow.ipynb
+++ b/docs/notebooks/nvidia/Llama_Stack_NVIDIA_E2E_Flow.ipynb
@@ -747,7 +747,8 @@
     "    benchmark_config={\n",
     "        \"eval_candidate\": {\n",
     "            \"type\": \"model\",\n",
-    "            \"model\": CUSTOMIZED_MODEL_DIR\n",
+    "            \"model\": CUSTOMIZED_MODEL_DIR,\n",
+    "            \"sampling_params\": {}\n",
     "        }\n",
     "    }\n",
     ")\n",
@@ -1281,7 +1282,8 @@
     "    benchmark_config={\n",
     "        \"eval_candidate\": {\n",
     "            \"type\": \"model\",\n",
-    "            \"model\": customized_chat_model_dir\n",
+    "            \"model\": customized_chat_model_dir,\n",
+    "            \"sampling_params\": {}\n",
     "        }\n",
     "    }\n",
     ")\n",