Update reproduction instruction
Browse files
README.md
CHANGED
|
@@ -132,9 +132,9 @@ llm = LLM(
|
|
| 132 |
)
|
| 133 |
```
|
| 134 |
|
| 135 |
-
### Training
|
| 136 |
|
| 137 |
-
|
| 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
|