def get_mp_merge_args(parser):
    """Provide extra arguments required for merging."""
    group = parser.add_argument_group(title='mp merge')

    group.add_argument('--model-type', type=str, required=True,
                       choices=['BERT', 'GPT2', 'RACE', 'MNLI', 'QQP'],
                       help='Type of the mdoel.')

    return parser
Provide extra arguments required for merging.
megatron_lm/tools/merge_mp_partitions.py