code: def get_files_to_download(n_layers: int = 44):
      """
      ### Get files to download
  
      :return: a list of files to be downloaded
      """
      layers = (
          # Embedding layer
          [0]
          +
          # Transformer layers
          list(range(2, 2 + n_layers))
          +
          # Final normalization layer and readout layer
          [47, 48]
      )
  
      return (
          # Vocabulary and configs
          ["20B_tokenizer.json", "configs/20B.yml", "latest"]
          +
          # Layer checkpoints
          [
              f"global_step150000/layer_{i :02d}-model_{p :02d}-model_states.pt"
              for i in layers
              for p in range(2)
          ]
          +
          # Empty states (not used)
          [
              f"global_step150000/mp_rank_{i :02d}_model_states.pt"
              for i in range(8)
          ]
      )
  
file_path: annotated_deep_learning_paper_implementations/labml_nn/neox/checkpoint.py
imports: from pathlib import Path
  from typing import Dict, Union, Tuple, Optional
  import torch
  from torch import nn
  from labml import monit, lab, logger
  from labml.logger import Text, inspect
  from labml.utils.download import download_file
label_desc: Use image node_id 2 for calling a global function with example usage: get_files_to_download(n_layers) and returns: unknown