class_docstr: Token-based (input/output is tokens) encoder-decoder model (e.g. T5 Language model.)
docstr: For easy load when model is combined with other heads,
  add an extra key.
docstr_tok: ["For","easy","load","when","model","is","combined","with","other","heads",",","add","an","extra","key","."]
code: def state_dict_for_save_checkpoint(
      self, destination=None, prefix="", keep_vars=False
  ):
      """For easy load when model is combined with other heads,
      add an extra key."""
  
      state_dict_ = {}
  
      state_dict_[
          self._encoder_embedding_key
      ] = self.encoder_embedding.state_dict_for_save_checkpoint(
          destination, prefix, keep_vars
      )
      state_dict_[
          self._decoder_embedding_key
      ] = self.decoder_embedding.state_dict_for_save_checkpoint(
          destination, prefix, keep_vars
      )
      state_dict_[
          self._enc_dec_model_key
      ] = self.enc_dec_model.state_dict_for_save_checkpoint(
          destination, prefix, keep_vars
      )
      state_dict_[
          self._tokens_head_key
      ] = self.tokens_head.state_dict_for_save_checkpoint(
          destination, prefix, keep_vars
      )
  
      return state_dict_
  
imports: import torch
  from omegaconf import DictConfig
  from nemo.collections.nlp.modules.common.megatron.adapters.parallel_adapters import AdapterName, PromptEncoderAdapterConfig
  from nemo.collections.nlp.modules.common.megatron.hiddens import get_hiddens_module
  from nemo.collections.nlp.modules.common.megatron.language_model import Embedding
  from nemo.collections.nlp.modules.common.megatron.layer_type import LayerType
  from nemo.collections.nlp.modules.common.megatron.megatron_decoders import get_decoder_model
  from nemo.collections.nlp.modules.common.megatron.megatron_encoder_decoder import MegatronTransformerEncoderDecoderModule
  from nemo.collections.nlp.modules.common.megatron.megatron_encoders import get_encoder_model
  from nemo.collections.nlp.modules.common.megatron.module import MegatronModule
  from nemo.collections.nlp.modules.common.megatron.position_embedding import ALiBiRelativePositionEmbedding, KERPLERelativePositionEmbedding, T5RelativePositionEmbedding
  from nemo.collections.nlp.modules.common.megatron.utils import ApexGuardDefaults, build_position_ids, init_method_normal, parallel_lm_logits, scaled_init_method_normal
  from nemo.collections.nlp.modules.common.megatron.vocab_parallel_cross_entropy import vocab_parallel_cross_entropy
  from nemo.core.classes.mixins import adapter_mixins
label_desc: Use image node_id 8 for calling the MegatronTokenLevelEncoderDecoderModule obj's underlying member method code with example usage: obj.state_dict_for_save_checkpoint(destination, prefix, keep_vars) and returns: state_dict_