class_docstr: T0 (https://arxiv.org/abs/2110.08207) Model that Inherits from MegatronT5SFTModel and overrides the dataset building.
code: def setup_training_dataloader(self):
      if hasattr(self, "_train_ds"):
          consumed_samples = self.compute_consumed_samples(0)
          self._train_dl = self.build_data_loader(
              dataset=self._train_ds,
              data_cfg=self.cfg.data.train_ds,
              consumed_samples=consumed_samples,
          )
  
imports: import torch
  from omegaconf import DictConfig, ListConfig
  from pytorch_lightning.trainer.trainer import Trainer
  from nemo.collections.nlp.data.language_modeling.megatron.base_dataset_utils import get_datasets_weights_and_num_samples
  from nemo.collections.nlp.data.language_modeling.megatron.blendable_dataset import BlendableDataset
  from nemo.collections.nlp.data.language_modeling.megatron.megatron_batch_samplers import MegatronPretrainingBatchSampler
  from nemo.collections.nlp.data.language_modeling.t0_dataset import T0Dataset
  from nemo.collections.nlp.models.language_modeling.megatron_t5_sft_model import MegatronT5SFTModel
  from nemo.utils import AppState, logging
label_desc: Use image node_id 8 for calling the MegatronT0Model obj's underlying member method code with example usage: obj.setup_training_dataloader() without return types