def METHOD_NAME(self) -> Optional[Dict[str, NeuralType]]:
    """Returns definitions of module output ports.
           """
    return {
        "input_ids": NeuralType(("B", "T"), ChannelType()),
        "attention_mask": NeuralType(("B", "T"), MaskType()),
        "decoder_input_ids": NeuralType(("B", "T"), ChannelType()),
        "lm_labels": NeuralType(("B", "T"), ChannelType()),
    }