code: def model_fn(model_dir):
      # load tokenizer and neuron model from model_dir
      tokenizer = AutoTokenizer.from_pretrained(model_dir)
      model = torch.jit.load(
          os.path.join(model_dir, AWS_NEURON_TRACED_WEIGHTS_NAME)
      )
      model_config = AutoConfig.from_pretrained(model_dir)
  
      return model, tokenizer, model_config
  
imports: import os
  from transformers import AutoConfig, AutoTokenizer
  import torch
  import torch.neuron
label_desc: Use image node_id 1 for calling a global function with example usage: model_fn(model_dir) and returns: model,  tokenizer,  model_config