code: def parse_args():
      parser = argparse.ArgumentParser()
      parser.add_argument("--model-name", type=str, help="model name")
      parser.add_argument(
          "--output-path",
          type=str,
          help="where to save the checkpoint",
      )
      parser.add_argument(
          "--dataset-path",
          type=str,
          help="location of the calibration dataset",
      )
      parser.add_argument("--num-samples", type=int, default=10)
      parser.add_argument("--seq-len", type=int, default=512)
      args = parser.parse_args()
      return args
  
file_path: ColossalAI/examples/inference/build_smoothquant_weight.py
imports: import argparse
  import os
  import torch
  from datasets import load_dataset
  from transformers import LlamaTokenizer
  from colossalai.inference.quant.smoothquant.models.llama import SmoothLlamaForCausalLM
label_desc: Use image node_id 2 for calling a global function with example usage: parse_args() and returns: args