Skip to content
Repo

convert_megatron_checkpoint_to_hf

from modal_dojo import convert_megatron_checkpoint_to_hf

Convert a Megatron checkpoint to Hugging Face format.

convert_megatron_checkpoint_to_hf(checkpoint: Checkpoint, model: ModelConfig, recipe: VllmRecipe | SglangRecipe | None = None) -> Checkpoint