from .pt import megatron_pt_main
from .sft import megatron_sft_main