Skip to content

Commit 0d79a3c

Browse files
h-guo18yeyu-nvidia
andauthored
adress comments
Co-authored-by: yeyu-nvidia <[email protected]> Signed-off-by: h-guo18 <[email protected]>
1 parent 2b02185 commit 0d79a3c

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

examples/speculative_decoding/example.ipynb

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -76,7 +76,7 @@
7676
"cell_type": "markdown",
7777
"metadata": {},
7878
"source": [
79-
"## Train Draft Head On Daring-Anteater\n",
79+
"## Train Draft module On Daring-Anteater\n",
8080
"We will fine-tune the draft head on the Daring-Anteater dataset using the standard Hugging Face Trainer. Note that only the draft model's weights are updated during this process; the original target model remains unchanged. After training, our speculative decoding model will be ready for export and deployment. Note that the time to train will be significantly dependent on the epochs (default=4) and the hardware being used."
8181
]
8282
},

0 commit comments

Comments
 (0)