code: def parse_args():
      parser = ArgumentParser()
      parser.add_argument(
          "--multi-gpu",
          default=False,
          action="store_true",
          help="Whether to use multi gpu.",
      )
      parser.add_argument(
          "--device-map",
          default=None,
          help="the device map for accelerate.",
      )
      parser.add_argument(
          "--delta", default=None, type=str, help="The delta path."
      )
      args = parser.parse_args()
      return args
  
file_path: DeepKE/example/llm/CPM-Bee/src/text_generation_hf.py
imports: import torch
  from collections import OrderedDict
  from argparse import ArgumentParser
  from transformers import AutoModelForCausalLM, AutoTokenizer
  from accelerate import dispatch_model
  from accelerate.utils import get_balanced_memory, infer_auto_device_map
label_desc: Use image node_id 1 for calling a global function with example usage: parse_args() and returns: args