code: def main():
      log.setup_main_logger(console=True, file_logging=False)
      params = argparse.ArgumentParser(
          description="Convert DeepSpeed checkpoints to regular parameter files in a Sockeye model directory."
      )
      params.add_argument(
          "--model",
          "-m",
          required=True,
          help="Model directory containing DeepSpeed checkpoints.",
      )
      params.add_argument(
          "--keep-deepspeed",
          "-k",
          action="store_true",
          help="Keep DeepSpeed checkpoints (renamed e.g., params.00000.ds).",
      )
      args = params.parse_args()
      convert_model_checkpoints(
          args.model, keep_deepspeed=args.keep_deepspeed
      )
  
file_path: sockeye/sockeye/convert_deepspeed.py
imports: import argparse
  import collections
  import gc
  import logging
  import os
  import shutil
  import torch
  from .None import constants
  from .None import log
  from .None import model
  from .None import utils
label_desc: Use image node_id 4 for calling a global function with example usage: main() without return types