code: def parse_args():
      parser = argparse.ArgumentParser(
          description="Benchmark for lightseq cpp inference"
      )
      parser.add_argument(
          "--batch_size", type=int, default=1, help="batch size"
      )
      parser.add_argument(
          "--beam_size",
          type=int,
          default=1,
          help="beam size in beam search",
      )
      parser.add_argument(
          "--input_seq_len",
          type=int,
          default=32,
          help="input sequence length",
      )
      parser.add_argument(
          "--output_seq_len",
          type=int,
          default=32,
          help="output sequence length",
      )
      parser.add_argument(
          "--model_name",
          type=str,
          default="facebook/bart-base",
          help="huggingface pretrain model or checkpoint path",
      )
      parser.add_argument(
          "--sampling_method",
          type=str,
          default="beam_search",
          help="generation method",
      )
      parser.add_argument(
          "--topk", type=int, default=4, help="topk when topk sampling"
      )
      parser.add_argument(
          "--topp",
          type=float,
          default=0.75,
          help="topp when topp sampling",
      )
      parser.add_argument(
          "--length_penalty",
          type=float,
          default=1.0,
          help="length penalty when generation",
      )
      parser.add_argument(
          "--enable_quant",
          type=bool,
          default=False,
          help="is quantized model",
      )
      args = parser.parse_args()
      return args
  
file_path: lightseq/examples/inference/python/generate_model.py
imports: import argparse
  import subprocess
  import os
  import re
  import h5py
  import numpy
  from export.huggingface.hf_bart_export import extract_transformer_weights
  from export.huggingface.hf_gpt2_export import extract_gpt_weights
  from export.huggingface.ls_hf_quant_gpt2_export import extract_gpt_weights
label_desc: Use image node_id 1 for calling a global function with example usage: parse_args() and returns: args