class_docstr: Actor model base class.
  
  Args:
      model (nn.Module): Actor Model.
      lora_rank (int): LoRA rank.
      lora_train_bias (str): LoRA bias training mode.
docstr: Returns model output.
docstr_tok: ["Returns","model","output","."]
code: def forward(
      self,
      input_ids: torch.LongTensor,
      attention_mask: Optional[torch.Tensor] = None,
      **model_kwargs,
  ) -> torch.Tensor:
      """Returns model output."""
      output = self.model(
          input_ids, attention_mask=attention_mask, **model_kwargs
      )
      return output
  
file_path: ColossalAI/applications/Chat/coati/models/base/actor.py
imports: from typing import Optional
  import torch
  import torch.nn
  from ..lora import LoRAModule
label_desc: Use image node_id 2 for calling the Actor obj's underlying member method code with example usage: obj.forward(input_ids, attention_mask) and returns: output