code: def forward(
      self, input_ids, token_type_ids=None, attention_mask=None
  ):
      # Only return lm_logits
      return self.model(
          input_ids=input_ids,
          token_type_ids=token_type_ids,
          attention_mask=attention_mask,
      )
  
file_path: colossalai/examples/community/roberta/pretraining/pretrain_utils.py
imports: import os
  import sys
  import torch
  import transformers
  from transformers import get_linear_schedule_with_warmup
  from colossalai.legacy.core import global_context
  from colossalai.nn.optimizer import HybridAdam
  from collections import OrderedDict
  import torch.nn
  from model.bert import BertForMaskedLM
  from model.deberta_v2 import DebertaV2ForMaskedLM
label_desc: Use image node_id 2 for calling the LMModel obj's underlying member method code with example usage: obj.forward(input_ids, token_type_ids, attention_mask) and returns: self