ms-swift/swift/megatron/train/__init__.py

from .pt import megatron_pt_main
from .sft import megatron_sft_main