address comments

h-guo18 · yeyu-nvidia · h-guo18 · commit 448d79c8d0cd · 2025-09-17T13:34:05.000-07:00
Co-authored-by: yeyu-nvidia &lt;yeyu@nvidia.com&gt;
Signed-off-by: h-guo18 &lt;67671475+h-guo18@users.noreply.github.com&gt;
diff --git a/examples/speculative_decoding/example.ipynb b/examples/speculative_decoding/example.ipynb
@@ -22,7 +22,7 @@
    "metadata": {},
    "source": [
     "## Convert Model for Speculative Decoding\n",
-    "Here, we'll adapt our base model for speculative decoding by attaching a smaller EAGLE Head. The upcoming code first loads meta-llama/Llama-3.2-1B as our base model and then configures the new draft head. To ensure compatibility, the draft head's dimensions must match the target model. Finally, the modelopt toolkit attaches this new, untrained head, leaving us with a combined model that is ready for the training phase later."
+    "Here, we'll adapt our base model for speculative decoding by attaching a smaller EAGLE module. The upcoming code first loads meta-llama/Llama-3.2-1B as our base model and then configures the new draft module. To ensure compatibility, the draft module's dimensions must match the target model. Finally, the modelopt toolkit attaches this new, untrained module, leaving us with a combined model that is ready for the training phase later."
    ]
   },
   {

Original file line number	Diff line number	Diff line change
`@@ -22,7 +22,7 @@`
`22`	`22`	`"metadata": {},`
`23`	`23`	`"source": [`
`24`	`24`	`"## Convert Model for Speculative Decoding\n",`
`25`		`- "Here, we'll adapt our base model for speculative decoding by attaching a smaller EAGLE Head. The upcoming code first loads meta-llama/Llama-3.2-1B as our base model and then configures the new draft head. To ensure compatibility, the draft head's dimensions must match the target model. Finally, the modelopt toolkit attaches this new, untrained head, leaving us with a combined model that is ready for the training phase later."`
	`25`	`+ "Here, we'll adapt our base model for speculative decoding by attaching a smaller EAGLE module. The upcoming code first loads meta-llama/Llama-3.2-1B as our base model and then configures the new draft module. To ensure compatibility, the draft module's dimensions must match the target model. Finally, the modelopt toolkit attaches this new, untrained module, leaving us with a combined model that is ready for the training phase later."`
`26`	`26`	`]`
`27`	`27`	`},`
`28`	`28`	`{`