code: def get_lightning_module(args):
      sp_model = spm.SentencePieceProcessor(
          model_file=str(args.sp_model_path)
      )
      if args.modality == "audiovisual":
          from lightning_av import AVConformerRNNTModule
  
          model = AVConformerRNNTModule(args, sp_model)
      else:
          from lightning import ConformerRNNTModule
  
          model = ConformerRNNTModule(args, sp_model)
      return model
  
file_path: audio/examples/avsr/train.py
imports: import logging
  import os
  from argparse import ArgumentParser
  import sentencepiece
  from average_checkpoints import ensemble
  from pytorch_lightning import seed_everything, Trainer
  from pytorch_lightning.callbacks import LearningRateMonitor, ModelCheckpoint
  from pytorch_lightning.strategies import DDPStrategy
  from transforms import get_data_module
label_desc: Use image node_id 2 for calling a global function with example usage: get_lightning_module(args) and returns: model