haoguo2000 commited on
Commit
fae607a
·
verified ·
1 Parent(s): 38a89ee

Update reproduction instruction

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -132,9 +132,9 @@ llm = LLM(
132
  )
133
  ```
134
 
135
- ### Training / Reproduction
136
 
137
- To reproduce this checkpoint — or train your own DFlash draft head — launch the streaming training recipe with [NVIDIA Model Optimizer](https://github.com/NVIDIA/Model-Optimizer):
138
 
139
  ```sh
140
  uv run launch.py --yaml examples/moonshotai/Kimi-K2.6/hf_streaming_dflash_multi_node.yaml --yes
 
132
  )
133
  ```
134
 
135
+ ### Training
136
 
137
+ See the following streaming recipe for [NVIDIA Model Optimizer](https://github.com/NVIDIA/Model-Optimizer) as a reference for training your own DFlash draft head:
138
 
139
  ```sh
140
  uv run launch.py --yaml examples/moonshotai/Kimi-K2.6/hf_streaming_dflash_multi_node.yaml --yes