code: def get_module_cls_by_model_name(model_cls_name):
      _module_by_model_name = {
          "Speech2Text2Decoder": "transformers.models.speech_to_text_2.modeling_speech_to_text_2",
          "TrOCRDecoder": "transformers.models.trocr.modeling_trocr",
      }
      module_name = _module_by_model_name.get(
          model_cls_name, "transformers"
      )
      module = importlib.import_module(module_name)
      return getattr(module, model_cls_name)
  
imports: import importlib
  import logging
  import os
  import re
  import subprocess
  import sys
  import warnings
  import torch
  from common import BenchmarkRunner, download_retry_decorator, main, reset_rng_state
  from torch._dynamo.testing import collect_results
  from torch._dynamo.utils import clone_inputs
label_desc: Use image node_id 2 for calling a global function with example usage: get_module_cls_by_model_name(model_cls_name) and returns: getattr