code: def parse_config():
      parser = argparse.ArgumentParser(description="arg parser")
      parser.add_argument(
          "--base_model",
          type=str,
          default="/data1/pretrained-models/llama-7b-hf",
      )
      parser.add_argument("--cache_dir", type=str, default="./cache")
      parser.add_argument(
          "--context_size",
          type=int,
          default=-1,
          help="context size during fine-tuning",
      )
      parser.add_argument(
          "--flash_attn", type=bool, default=True, help=""
      )
      parser.add_argument(
          "--temperature", type=float, default=0.6, help=""
      )
      parser.add_argument("--top_p", type=float, default=0.9, help="")
      parser.add_argument(
          "--max_gen_len", type=int, default=512, help=""
      )
      parser.add_argument("--host", type=str, default="localhost")
      parser.add_argument("--port", type=int, default=8898)
      args = parser.parse_args()
      return args
  
file_path: long-lora/demo.py
imports: import os
  import sys
  import math
  import torch
  import argparse
  import textwrap
  import transformers
  from peft import PeftModel
  from transformers import GenerationConfig, TextIteratorStreamer
  from llama_attn_replace import replace_llama_attn
  from threading import Thread
  import gradio
label_desc: Use image node_id 1 for calling a global function with example usage: parse_config() and returns: args