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 = SglangRecipe(_recipe_type=<DeployRecipeType.SGLANG: sglang>, gpu=H100, tp=None, dp=None, context_length=None, mem_fraction_static=None, chunked_prefill_size=None, max_running_requests=None, sglang_image='lmsysorg/sglang:v0.5.12', extra_server_args=None, env_vars={}, install_transformers_from_git=True, environment_name=None, deploy_strategy='rolling', startup_timeout=1200)) -> Checkpoint