code: def set_args():
      parser = argparse.ArgumentParser()
      parser.add_argument(
          "--test_path", default="data/spo_1.json", type=str, help=""
      )
      parser.add_argument("--device", default="3", type=str, help="")
      parser.add_argument(
          "--model_dir", default="/data/model", type=str, help=""
      )
      parser.add_argument("--max_len", type=int, default=768, help="")
      parser.add_argument(
          "--max_src_len", type=int, default=450, help=""
      )
      parser.add_argument(
          "--prompt_text", type=str, default="", help=""
      )
      return parser.parse_args()
  
file_path: DeepKE/example/llm/InstructKGC/src/inference_pt.py
imports: import torch
  import json
  from tqdm import tqdm
  import time
  import os
  import argparse
  from transformers import AutoTokenizer, AutoModel
label_desc: Use image node_id 1 for calling a global function with example usage: set_args() and returns: parser