def METHOD_NAME(cfg):
    if cfg.model_type == 'gpt':
        return MegatronGPTModel
    elif cfg.model_type == 'bert':
        return MegatronBertModel
    elif cfg.model_type == 't5':
        return MegatronT5Model
    elif cfg.model_type == 'bart':
        return MegatronBARTModel
    elif cfg.model_type == 'nmt':
        return MegatronNMTModel
    elif cfg.model_type == 'retro':
        return MegatronRetrievalModel
    else:
        raise ValueError("Invalid Model Type")