code: def get_model_name(model_name):
      model_name = model_name.lower()
      for key, values in MODEL_DICT.items():
          for v in values:
              if v == model_name:
                  return key
      return "other"
  
file_path: DeepKE/example/llm/InstructKGC/src/utils/general_utils.py
imports: import os
  import random
  import time
  import numpy
  import torch
  from transformers import AutoTokenizer, AutoModel, AutoModelForCausalLM, LlamaForCausalLM, LlamaTokenizer, Trainer, Seq2SeqTrainer
label_desc: Use image node_id 2 for calling a global function with example usage: get_model_name(model_name) and returns: str, key