Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/bytedance/MegaTTS3
/ types & classes
Types & classes
69 in github.com/bytedance/MegaTTS3
⨍
Functions
298
◇
Types & classes
69
↓ 8 callers
Class
Conv1d
tts/modules/aligner/whisper_small.py:56
↓ 8 callers
Class
LayerNorm
Layer normalization module. :param int nout: output dim size :param int dim: dimension to be normalized
tts/modules/ar_dur/commons/layers.py:19
↓ 8 callers
Class
Linear
tts/modules/aligner/whisper_small.py:47
↓ 5 callers
Class
LayerNorm
tts/modules/aligner/whisper_small.py:42
↓ 5 callers
Class
LayerNorm
tts/modules/ar_dur/commons/rel_transformer.py:279
↓ 5 callers
Class
SConv1d
tts/modules/wavvae/encoder/common_modules/conv.py:121
↓ 3 callers
Class
MultiheadAttention
tts/modules/ar_dur/commons/transformer.py:152
↓ 3 callers
Class
ResBlockMRFV2
tts/modules/wavvae/decoder/hifigan_modules.py:174
↓ 3 callers
Class
SinusoidalPositionalEmbedding
This module produces sinusoidal positional embeddings of any length. Padding symbols are ignored.
tts/modules/ar_dur/commons/transformer.py:27
↓ 3 callers
Class
TransformerFFNLayer
tts/modules/ar_dur/commons/transformer.py:89
↓ 2 callers
Class
MegaTTS3DiTInfer
tts/infer_cli.py:72
↓ 2 callers
Class
MultiHeadAttention
tts/modules/aligner/whisper_small.py:84
↓ 2 callers
Class
PosEmb
tts/modules/ar_dur/commons/nar_tts_modules.py:61
↓ 2 callers
Class
RelTransformerEncoder
tts/modules/ar_dur/commons/rel_transformer.py:335
↓ 2 callers
Class
ResidualAttentionBlock
tts/modules/aligner/whisper_small.py:134
↓ 2 callers
Class
RotaryEmbeddings
tts/modules/ar_dur/commons/rot_transformer.py:91
↓ 2 callers
Class
TokenTextEncoder
Encoder based on a user-supplied vocabulary (file or list).
tts/utils/text_utils/text_encoder.py:121
↓ 2 callers
Class
Upsample
tts/modules/wavvae/decoder/hifigan_modules.py:35
↓ 1 callers
Class
ARDurPredictor
tts/modules/ar_dur/ar_dur_predictor.py:171
↓ 1 callers
Class
AdaLNZero
tts/modules/llm_dit/transformer.py:52
↓ 1 callers
Class
AdaLNZero_Out
tts/modules/llm_dit/transformer.py:66
↓ 1 callers
Class
Args
tts/utils/commons/hparams.py:26
↓ 1 callers
Class
Attention
tts/modules/llm_dit/transformer.py:80
↓ 1 callers
Class
AudioEncoder
tts/modules/aligner/whisper_small.py:168
↓ 1 callers
Class
ConditionalFlowMatcher
Base class for conditional flow matching methods. This class implements the independent conditional flow matching methods from [1] and serves as a
tts/modules/llm_dit/cfm.py:71
↓ 1 callers
Class
ConvLayerNorm
tts/modules/wavvae/encoder/common_modules/conv.py:96
↓ 1 callers
Class
ConvReluNorm
tts/modules/ar_dur/commons/rel_transformer.py:300
↓ 1 callers
Class
DecSALayer
tts/modules/ar_dur/commons/transformer.py:543
↓ 1 callers
Class
DiagonalGaussianDistribution
tts/modules/wavvae/decoder/diag_gaussian.py:18
↓ 1 callers
Class
Diffusion
tts/modules/llm_dit/dit.py:27
↓ 1 callers
Class
EncSALayer
tts/modules/ar_dur/commons/transformer.py:500
↓ 1 callers
Class
Encoder
tts/modules/ar_dur/commons/rel_transformer.py:41
↓ 1 callers
Class
Encoder
tts/modules/wavvae/decoder/seanet_encoder.py:21
↓ 1 callers
Class
FFN
tts/modules/ar_dur/commons/rel_transformer.py:254
↓ 1 callers
Class
FeedForward
tts/modules/llm_dit/transformer.py:130
↓ 1 callers
Class
Generator
tts/modules/wavvae/decoder/hifigan_modules.py:228
↓ 1 callers
Class
LengthRegulator
tts/modules/ar_dur/commons/nar_tts_modules.py:23
↓ 1 callers
Class
LogitNormalTrainingTimesteps
tts/modules/llm_dit/cfm.py:36
↓ 1 callers
Class
MultiHeadAttention
tts/modules/ar_dur/commons/rel_transformer.py:100
↓ 1 callers
Class
NormConv1d
tts/modules/wavvae/encoder/common_modules/conv.py:107
↓ 1 callers
Class
ResBlockMRFV2Inter
tts/modules/wavvae/decoder/hifigan_modules.py:213
↓ 1 callers
Class
RotDecSALayer
tts/modules/ar_dur/commons/rot_transformer.py:543
↓ 1 callers
Class
RotDecSALayer2
tts/modules/ar_dur/commons/rot_transformer.py:612
↓ 1 callers
Class
RotMultiheadAttention
tts/modules/ar_dur/commons/rot_transformer.py:206
↓ 1 callers
Class
RotMultiheadAttention2
tts/modules/ar_dur/commons/rot_transformer.py:404
↓ 1 callers
Class
RotTransformerDecoderLayer
tts/modules/ar_dur/commons/rot_transformer.py:622
↓ 1 callers
Class
SEANetEncoder
tts/modules/wavvae/encoder/common_modules/seanet.py:72
↓ 1 callers
Class
SEANetResnetBlock
tts/modules/wavvae/encoder/common_modules/seanet.py:41
↓ 1 callers
Class
SLSTM
LSTM without worrying about the hidden state, nor the layout of the data. Expects input as convolutional layout.
tts/modules/wavvae/encoder/common_modules/lstm.py:34
↓ 1 callers
Class
SinusPositionEmbedding
tts/modules/llm_dit/time_embedding.py:20
↓ 1 callers
Class
TextDecoder
tts/modules/aligner/whisper_small.py:201
↓ 1 callers
Class
TimestepEmbedding
tts/modules/llm_dit/time_embedding.py:34
↓ 1 callers
Class
Transformer
tts/modules/llm_dit/transformer.py:207
↓ 1 callers
Class
TransformerBlock
tts/modules/llm_dit/transformer.py:154
↓ 1 callers
Class
TransformerEncoderLayer
tts/modules/ar_dur/commons/transformer.py:639
↓ 1 callers
Class
WavVAE_V3
tts/modules/wavvae/decoder/wavvae_v3.py:25
↓ 1 callers
Class
Whisper
tts/modules/aligner/whisper_small.py:246
Class
Audio2Mel
tts/modules/wavvae/decoder/hifigan_modules.py:99
Class
CodePredictor
tts/modules/ar_dur/ar_dur_predictor.py:48
Class
Downsample
tts/modules/wavvae/decoder/hifigan_modules.py:59
Class
FFTBlocks
tts/modules/ar_dur/commons/transformer.py:676
Class
FastSpeechEncoder
tts/modules/ar_dur/commons/transformer.py:734
Class
Permute
tts/modules/ar_dur/commons/layers.py:50
Class
Reshape
tts/modules/ar_dur/commons/layers.py:41
Class
ResnetBlock
tts/modules/wavvae/decoder/hifigan_modules.py:149
Class
SinusoidalPositionalEmbedding
This module produces sinusoidal positional embeddings of any length. Padding symbols are ignored.
tts/modules/ar_dur/commons/rot_transformer.py:29
Class
TextEncoder
Base class for converting from ints to/from human readable strings.
tts/utils/text_utils/text_encoder.py:52
Class
TransformerDecoderLayer
tts/modules/ar_dur/commons/transformer.py:654
Class
VariancePreservingConditionalFlowMatcher
Albergo et al. 2023 trigonometric interpolants class. This class inherits the ConditionalFlowMatcher and override the compute_mu_t and compute_con
tts/modules/llm_dit/cfm.py:252