code: def _prepare_for_export(self, **kwargs):
      self.model.remove_norms()
      super()._prepare_for_export(**kwargs)
  
      tensor_shape = (
          ("T")
          if self.export_config["enable_ragged_batches"]
          else ("B", "T")
      )
  
      # Define input_types and output_types as required by export()
      self._input_types = {
          "text": NeuralType(tensor_shape, TokenIndex()),
          "batch_lengths": NeuralType(("B")),
          "speaker_id": NeuralType(("B"), Index()),
          "speaker_id_text": NeuralType(("B"), Index()),
          "speaker_id_attributes": NeuralType(("B"), Index()),
          "pitch": NeuralType(tensor_shape, RegressionValuesType()),
          "pace": NeuralType(tensor_shape),
      }
      self._output_types = {
          "spect": NeuralType(
              ("B", "D", "T_spec"), MelSpectrogramType()
          ),
          "num_frames": NeuralType(("B"), TokenDurationType()),
          "durs_predicted": NeuralType(
              ("B", "T_text"), TokenDurationType()
          ),
      }
      if self.export_config["enable_volume"]:
          self._input_types["volume"] = NeuralType(
              tensor_shape, optional=True
          )
          self._output_types["volume_aligned"] = NeuralType(
              ("B", "T_spec"), RegressionValuesType()
          )
  
imports: import contextlib
  import torch
  from hydra.utils import instantiate
  from omegaconf import DictConfig, OmegaConf
  from pytorch_lightning import Trainer
  from pytorch_lightning.loggers import TensorBoardLogger
  from nemo.collections.common.tokenizers.text_to_speech.tts_tokenizers import BaseTokenizer
  from nemo.collections.tts.losses.radttsloss import AttentionBinarizationLoss, RADTTSLoss
  from nemo.collections.tts.models.base import SpectrogramGenerator
  from nemo.collections.tts.parts.utils.helpers import batch_from_ragged, g2p_backward_compatible_support, plot_alignment_to_numpy, regulate_len, sample_tts_input
  from nemo.core.classes import Exportable
  from nemo.core.classes.common import typecheck
  from nemo.core.neural_types.elements import Index, MelSpectrogramType, RegressionValuesType, TokenDurationType, TokenIndex
  from nemo.core.neural_types.neural_type import NeuralType
  from nemo.core.optim.radam import RAdam
  from nemo.utils import logging
  from nemo.utils.decorators import experimental
label_desc: Use image node_id 20 for calling the RadTTSModel obj's underlying member method code with example usage: obj._prepare_for_export() without return types