code: def parse_arguments():
      parser = argparse.ArgumentParser()
      parser.add_argument(
          "--input_folder",
          type=str,
          required=True,
          help="Input DeepSpeed Checkpoint folder",
      )
      parser.add_argument(
          "--output_folder",
          type=str,
          required=True,
          help="Output DeepSpeed checkpoint folder",
      )
      parser.add_argument(
          "--num_extract_workers",
          default=4,
          type=int,
          help="How many parallel processes to extract zero shards",
      )
      parser.add_argument(
          "--num_merge_workers",
          default=2,
          type=int,
          help="How many parallel processes to merge tp slices (more memory intensive, use much fewer than --num_extract_workers))",
      )
      parser.add_argument(
          "--keep_temp_folder",
          action="store_true",
          help="Preserve temporary folder of intermediate checkpoint slice files. Useful for debugging.",
      )
      parser.add_argument(
          "--no_strict",
          dest="strict",
          action="store_false",
          help="Do not perform validity checks on converted checkpoint.",
      )
      args = parser.parse_args()
      print(f"args = {args}")
      return args
  
imports: from functools import partial
  import argparse
  import glob
  import itertools
  import multiprocessing
  import os
  import re
  import shutil
  import torch
  import tqdm
  from deepspeed.checkpoint import DeepSpeedCheckpoint
  from deepspeed.checkpoint import OPTIMIZER_STATE_DICT, BASE_OPTIMIZER_STATE, SINGLE_PARTITION_OF_FP32_GROUPS, PARAM_SLICE_MAPPINGS, PARAM_SHAPES, PARAM, CAT_DIM, PARAM_N_SUB_PARAMS, VOCAB_TENSOR, UNIVERSAL_CHECKPOINT_INFO, VOCABULARY_PARAMETER_PATTERNS, PIPELINE_REPLICATED_PARAMETER_PATTERNS, TP_REPLICATED_PARAMETER_PATTERNS, PARAMETER_TO_AVERAGE_PATTERNS, PARAMETER_WITH_ROW_PARALLELISM_PATTERNS, PARAMETER_WITH_2_SUB_PARAMS_CAT_DIM_0
label_desc: Use image node_id 1 for calling a global function with example usage: parse_arguments() and returns: args