code: def model(model_name):
      model = T5.from_pretrained_module(model_name).eval()
      model.beam_search.max_steps = 5
      return model
  
file_path: allennlp/tests/modules/transformer/t5_test.py
imports: import pytest
  import torch
  from transformers.models import t5
  from allennlp.modules.transformer.t5 import T5
  from allennlp.nn.parallel import FairScaleFsdpAccelerator
  from allennlp.common.testing import run_distributed_test, requires_multi_gpu
label_desc: Use image node_id 3 for calling a global function with example usage: model(model_name) and returns: model