code: def setup_method(self, method):
      Utils.initialize_model_parallel(1, 1)
      model_parallel_cuda_manual_seed(123)
      transformer_config = TransformerConfig(
          num_layers=12,
          hidden_size=768,
          num_attention_heads=12,
          kv_channels=64,
          ffn_hidden_size=3072,
          use_cpu_initialization=True,
      )
      en_block_spec = get_t5_encoder_with_transformer_engine_block_spec(
          12
      )
      de_block_spec = get_t5_decoder_with_transformer_engine_block_spec(
          12
      )
      self.t5_model = T5Model(
          config=transformer_config,
          transformer_encoder_layer_spec=en_block_spec,
          transformer_decoder_layer_spec=de_block_spec,
          vocab_size=29184,
          max_sequence_length=4,
      )
  
file_path: megatron-lm/tests/unit_tests/models/test_t5_model.py
imports: import pytest
  import torch
  from megatron.core.transformer.transformer_config import TransformerConfig
  from megatron.core.models.T5.t5_model import T5Model
  from tests.unit_tests.test_utilities import Utils
  from megatron.core.tensor_parallel.random import model_parallel_cuda_manual_seed
  from megatron.core.models.T5.t5_spec import get_t5_encoder_with_transformer_engine_block_spec, get_t5_decoder_with_transformer_engine_block_spec, get_t5_encoder_with_local_block_spec, get_t5_decoder_with_local_block_spec
label_desc: Use image node_id 1 for calling the TestT5Model obj's underlying member method code with example usage: obj.setup_method(method) without return types