def METHOD_NAME(self, method):
    Utils.initialize_model_parallel(1,1)
    model_parallel_cuda_manual_seed(123)
    self.transformer_config = TransformerConfig(num_layers=2, hidden_size=12, num_attention_heads=4, use_cpu_initialization=True)
    self.parallel_transformer_block = TransformerBlock(self.transformer_config)