class_docstr: GPT-2 Language model.
class_docstr_tok: ["GPT-2","Language","model","."]
docstr: See megatron.model.transformer.set_input_tensor()
docstr_tok: ["See","megatron.model.transformer.set_input_tensor","(",")"]
code: def set_input_tensor(self, input_tensor):
      """See megatron.model.transformer.set_input_tensor()"""
      self.language_model.set_input_tensor(input_tensor)
  
file_path: apex/apex/transformer/testing/standalone_gpt.py
imports: import contextlib
  import torch
  from apex.transformer.enums import AttnMaskType
  from apex.transformer.enums import ModelType
  from apex.transformer import tensor_parallel
  from apex.transformer.testing.global_vars import get_args
  from apex.transformer.testing.standalone_transformer_lm import MegatronModule
  from apex.transformer.testing.standalone_transformer_lm import parallel_lm_logits
  from apex.transformer.testing.standalone_transformer_lm import post_language_model_processing
  from apex.transformer.testing.standalone_transformer_lm import get_language_model
  from apex.transformer.testing.standalone_transformer_lm import init_method_normal
  from apex.transformer.testing.standalone_transformer_lm import scaled_init_method_normal
label_desc: Use image node_id 2 for calling the GPTModel obj's underlying member method code with example usage: obj.set_input_tensor(input_tensor) without return types