code: def parse_args():
      parser = argparse.ArgumentParser()
      parser.add_argument(
          "--model_name_or_path",
          type=str,
          default="models/llama/llama-7b",
      )
      parser.add_argument("--revision", type=str, default="main")
      parser.add_argument(
          "--tokenizer_name_or_path", type=str, default=None
      )
      parser.add_argument(
          "--dataset_name", type=str, default="wikitext"
      )
  
      parser.add_argument(
          "--task", type=str, default="wikitext-2-raw-v1"
      )
      parser.add_argument(
          "--split",
          type=str,
          default="test",
          choices=["validation", "test"],
      )
  
      parser.add_argument(
          "--num_samples",
          type=int,
          default=1,
      )
  
      parser.add_argument(
          "--output_dir",
          type=str,
          default="outputs/debug",
      )
  
      parser.add_argument(
          "--enable_start_recent_kv_cache", action="store_true"
      )
      parser.add_argument("--start_size", type=int, default=1)
      parser.add_argument("--recent_size", type=int, default=255)
      parser.add_argument("--enable_pos_shift", action="store_true")
  
      parser.add_argument("--num_eval_tokens", type=int, default=None)
  
      args = parser.parse_args()
      return args
  
file_path: long-lora/streaming_llm/utils.py
imports: import torch
  import argparse
  from transformers import AutoTokenizer, AutoModelForCausalLM
  import os.path
  import ssl
  import urllib.request
  import os
  import json
label_desc: Use image node_id 1 for calling a global function with example usage: parse_args() and returns: args