Checkpoint Conversion Steps#

This section documents the conversion steps registered under src/nemotron/steps/convert/. Use these steps to bridge checkpoint layouts between Hugging Face, Megatron distributed checkpoints, and LoRA adapter outputs.

Steps#

convert/hf_to_megatron

Import a Hugging Face checkpoint or model id into Megatron distributed checkpoint layout.

reference

convert/hf_to_megatron
convert/megatron_to_hf

Export a Megatron distributed checkpoint iteration into Hugging Face safetensors layout.

reference

convert/megatron_to_hf
convert/merge_lora

Merge a LoRA adapter with its original base checkpoint to produce a standalone checkpoint.

reference

convert/merge_lora