code: def get_megatron_config(pretrained_model_name: str) -> Dict[str, int]:
      """
      Returns Megatron-LM model config file
  
      Args:
          pretrained_model_name (str): pretrained model name
  
      Returns:
          config (dict): contains model configuration: number of hidden layers, number of attention heads, etc
      """
      _check_megatron_name(pretrained_model_name)
      return MEGATRON_CONFIG_MAP[pretrained_model_name]["config"]
  
file_path: NeMo/nemo/collections/nlp/modules/common/megatron/megatron_utils.py
imports: import os
  import shutil
  from typing import Dict, List
  import torch
  import wget
  from torch.hub import _get_torch_home
  from nemo.utils import logging
label_desc: Use image node_id 4 for calling a global function with example usage: get_megatron_config(pretrained_model_name) and returns: MEGATRON_CONFIG_MAP