Skip to content

Commit c249efc

Browse files
authored
docs: fix checkpointing command for megatron->hf export (#823)
Signed-off-by: abdalgader-a <abdalgader.abubaker@tii.ae>
1 parent 9389dfb commit c249efc

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

README.md

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -413,13 +413,13 @@ uv run python examples/converters/convert_dcp_to_hf.py \
413413
--hf-ckpt-path results/grpo/hf
414414
```
415415

416-
If you have a model saved in Megatron format, you can use the following command to convert it to Hugging Face format prior to running evaluation:
416+
If you have a model saved in Megatron format, you can use the following command to convert it to Hugging Face format prior to running evaluation. This script requires mcore, so make sure to launch with the mcore extra:
417417

418418
```sh
419419
# Example for a GRPO checkpoint at step 170
420-
uv run python examples/converters/convert_megatron_to_hf.py \
420+
uv run --extra mcore python examples/converters/convert_megatron_to_hf.py \
421421
--config results/grpo/step_170/config.yaml \
422-
--dcp-ckpt-path results/grpo/step_170/policy/weights/iter_0000000 \
422+
--megatron-ckpt-path results/grpo/step_170/policy/weights/iter_0000000 \
423423
--hf-ckpt-path results/grpo/hf
424424
```
425425

0 commit comments

Comments
 (0)