code: def get_linear_embedding_layers(model_type):
      """
      returns the linear embedding layers needed for loras, dependent on the model arch
      """
      if model_type == "phi-msft":
          return ["embd.wte", "lm_head.linear"]
      if model_type == "gpt_neox":
          return ["embed_in", "embed_out"]
      return ["embed_tokens", "lm_head"]
  
label_desc: Use image node_id 1 for calling a global function with example usage: get_linear_embedding_layers(model_type) and returns: list, list, list