SpeechT5/SpeechUT/speechut/modules/__init__.py

# --------------------------------------------------------
# Copyright (c) 2022 Microsoft
# Licensed under The MIT License [see LICENSE for details]
# Based on fairseq code bases
# https://github.com/facebookresearch/fairseq
# --------------------------------------------------------

from .learned_positional_embedding import LearnedPositionalEmbedding
from .multihead_attention import MultiheadAttention
from .relative_pos_enc import RelativePositionalEncoding
from .transformer_layer import TransformerEncoderLayerBase, TransformerDecoderLayerBase
from .w2v_encoder import TransformerEncoder, TransformerSentenceEncoderLayer
from .transformer_encoder import TransformerEncoderBase
from .transformer_decoder import TransformerDecoderScriptable, TransformerDecoderBaseScriptable

__all__ = [
    "MultiheadAttention",
    "RelativePositionalEncoding",
    "LearnedPositionalEmbedding",
    "TransformerEncoderLayerBase",
    "TransformerDecoderLayerBase",
    "TransformerEncoder",
    "TransformerSentenceEncoderLayer",
    "TransformerEncoderBase",
    "TransformerDecoderScriptable",
    "TransformerDecoderBaseScriptable",
]