SYMBOL INDEX (373 symbols across 29 files)

FILE: megatts3node.py
  function get_all_files (line 35) | def get_all_files(
  function get_speakers (line 89) | def get_speakers():
  class MegaTTS3DiTInfer (line 97) | class MegaTTS3DiTInfer():
    method __init__ (line 98) | def __init__(
    method clean (line 141) | def clean(self):
    method build_model (line 161) | def build_model(self, device):
    method preprocess (line 232) | def preprocess(self, audio_bytes, latent_file=None, topk_dur=1, **kwar...
    method forward (line 277) | def forward(self, texts, time_step, p_w, t_w, dur_disturb=0.1, dur_alp...
  class MegaTTS3SpeakersPreview (line 334) | class MegaTTS3SpeakersPreview:
    method INPUT_TYPES (line 336) | def INPUT_TYPES(s):
    method preview (line 346) | def preview(self, speaker):
  function cache_audio_tensor (line 361) | def cache_audio_tensor(
  function statistical_compare (line 384) | def statistical_compare(tensor1, tensor2):
  class MegaTTS3Run (line 402) | class MegaTTS3Run:
    method __init__ (line 403) | def __init__(self):
    method INPUT_TYPES (line 409) | def INPUT_TYPES(s):
    method clone (line 431) | def clone(self, audio, text, time_step, p_w, t_w, unload_model, audio_...
    method get_speaker_text_audio (line 511) | def get_speaker_text_audio(self, text, audio_1, audio_2):
  class MultiLinePromptMG (line 531) | class MultiLinePromptMG:
    method INPUT_TYPES (line 533) | def INPUT_TYPES(cls):
    method promptgen (line 548) | def promptgen(self, multi_line_prompt: str):

FILE: tts/frontend_function.py
  function g2p (line 24) | def g2p(self, text_inp):
  function align (line 40) | def align(self, wav):
  function make_dur_prompt (line 77) | def make_dur_prompt(self, mel2ph_ref, ph_ref, tone_ref):
  function dur_pred (line 95) | def dur_pred(self, ctx_dur_tokens, incremental_state_dur_prompt, ph_pred...
  function prepare_inputs_for_dit (line 154) | def prepare_inputs_for_dit(self, mel2ph_ref, mel2ph_pred, ph_ref, tone_r...

FILE: tts/modules/aligner/whisper_small.py
  class LayerNorm (line 42) | class LayerNorm(nn.LayerNorm):
    method forward (line 43) | def forward(self, x: Tensor) -> Tensor:
  class Linear (line 47) | class Linear(nn.Linear):
    method forward (line 48) | def forward(self, x: Tensor) -> Tensor:
  class Conv1d (line 56) | class Conv1d(nn.Conv1d):
    method _conv_forward (line 57) | def _conv_forward(
  function sinusoids (line 65) | def sinusoids(length, channels, max_timescale=10000):
  function disable_sdpa (line 75) | def disable_sdpa():
  class MultiHeadAttention (line 84) | class MultiHeadAttention(nn.Module):
    method __init__ (line 87) | def __init__(self, n_state: int, n_head: int):
    method forward (line 95) | def forward(
    method qkv_attention (line 118) | def qkv_attention(
  class ResidualAttentionBlock (line 134) | class ResidualAttentionBlock(nn.Module):
    method __init__ (line 135) | def __init__(self, n_state: int, n_head: int, cross_attention: bool = ...
    method forward (line 152) | def forward(
  class AudioEncoder (line 168) | class AudioEncoder(nn.Module):
    method __init__ (line 169) | def __init__(
    method forward (line 182) | def forward(self, x: Tensor, attn_mask: Tensor):
  class TextDecoder (line 201) | class TextDecoder(nn.Module):
    method __init__ (line 202) | def __init__(
    method forward (line 220) | def forward(self, x: Tensor, attn_mask: Tensor, xa: Tensor, kv_cache: ...
  class Whisper (line 246) | class Whisper(nn.Module):
    method __init__ (line 247) | def __init__(self):
    method embed_audio (line 261) | def embed_audio(self, mel: torch.Tensor):
    method logits (line 264) | def logits(self, tokens, audio_features, kv_cache=None):
    method forward (line 267) | def forward(
    method device (line 275) | def device(self):
    method install_kv_cache_hooks (line 278) | def install_kv_cache_hooks(self, cache: Optional[dict] = None):
    method sequence_mask (line 311) | def sequence_mask(self, seq_lens, max_len=None, device='cpu'):

FILE: tts/modules/ar_dur/ar_dur_predictor.py
  function fill_with_neg_inf2 (line 37) | def fill_with_neg_inf2(t):
  function expand_states (line 41) | def expand_states(h, mel2token):
  class CodePredictor (line 48) | class CodePredictor(nn.Module):
    method __init__ (line 49) | def __init__(self, hparams, hidden_size, dec_hidden_size, lm_num_layer...
    method forward_ling_encoder (line 87) | def forward_ling_encoder(
    method sample_one_step (line 130) | def sample_one_step(self, vq_pred):
    method forward_style_embed (line 148) | def forward_style_embed(self, spk_embed=None, spk_id=None, mel_ref=None):
    method buffered_future_mask (line 159) | def buffered_future_mask(self, tensor):
  class ARDurPredictor (line 171) | class ARDurPredictor(CodePredictor):
    method __init__ (line 172) | def __init__(self, hparams, hidden_size, dec_hidden_size, lm_num_layer...
    method forward (line 190) | def forward(self, txt_tokens, ling_feas, char_tokens, ph2char, bert_em...
    method infer (line 265) | def infer(self, txt_tokens, ling_feas, char_tokens, ph2char, bert_embed,
    method streaming_infer (line 322) | def streaming_infer(self, txt_tokens, ling_feas, char_tokens, ph2char,...

FILE: tts/modules/ar_dur/commons/layers.py
  class LayerNorm (line 19) | class LayerNorm(torch.nn.LayerNorm):
    method __init__ (line 25) | def __init__(self, nout, dim=-1, eps=1e-5):
    method forward (line 30) | def forward(self, x):
  class Reshape (line 41) | class Reshape(nn.Module):
    method __init__ (line 42) | def __init__(self, *args):
    method forward (line 46) | def forward(self, x):
  class Permute (line 50) | class Permute(nn.Module):
    method __init__ (line 51) | def __init__(self, *args):
    method forward (line 55) | def forward(self, x):
  function Embedding (line 59) | def Embedding(num_embeddings, embedding_dim, padding_idx=None):

FILE: tts/modules/ar_dur/commons/nar_tts_modules.py
  class LengthRegulator (line 23) | class LengthRegulator(torch.nn.Module):
    method __init__ (line 24) | def __init__(self, pad_value=0.0):
    method forward (line 28) | def forward(self, dur, dur_padding=None, alpha=1.0):
  class PosEmb (line 61) | class PosEmb(nn.Module):
    method __init__ (line 62) | def __init__(self, dim):
    method forward (line 70) | def forward(self, x):

FILE: tts/modules/ar_dur/commons/rel_transformer.py
  function convert_pad_shape (line 23) | def convert_pad_shape(pad_shape):
  function shift_1d (line 29) | def shift_1d(x):
  function sequence_mask (line 34) | def sequence_mask(length, max_length=None):
  class Encoder (line 41) | class Encoder(nn.Module):
    method __init__ (line 42) | def __init__(self, hidden_channels, filter_channels, n_heads, n_layers...
    method forward (line 71) | def forward(self, x, x_mask, attn_mask=1):
  class MultiHeadAttention (line 100) | class MultiHeadAttention(nn.Module):
    method __init__ (line 101) | def __init__(self, channels, out_channels, n_heads, window_size=None, ...
    method forward (line 135) | def forward(self, x, c, attn_mask=None):
    method attention (line 145) | def attention(self, query, key, value, mask=None):
    method _matmul_with_relative_values (line 178) | def _matmul_with_relative_values(self, x, y):
    method _matmul_with_relative_keys (line 187) | def _matmul_with_relative_keys(self, x, y):
    method _get_relative_embeddings (line 196) | def _get_relative_embeddings(self, relative_embeddings, length):
    method _relative_position_to_absolute_position (line 211) | def _relative_position_to_absolute_position(self, x):
    method _absolute_position_to_relative_position (line 228) | def _absolute_position_to_relative_position(self, x):
    method _attention_bias_proximal (line 242) | def _attention_bias_proximal(self, length):
  class FFN (line 254) | class FFN(nn.Module):
    method __init__ (line 255) | def __init__(self, in_channels, out_channels, filter_channels, kernel_...
    method forward (line 268) | def forward(self, x, x_mask):
  class LayerNorm (line 279) | class LayerNorm(nn.Module):
    method __init__ (line 280) | def __init__(self, channels, eps=1e-4):
    method forward (line 288) | def forward(self, x):
  class ConvReluNorm (line 300) | class ConvReluNorm(nn.Module):
    method __init__ (line 301) | def __init__(self, in_channels, hidden_channels, out_channels, kernel_...
    method forward (line 325) | def forward(self, x, x_mask):
  class RelTransformerEncoder (line 335) | class RelTransformerEncoder(nn.Module):
    method __init__ (line 336) | def __init__(self,
    method forward (line 387) | def forward(self, x, x_mask=None, other_embeds=0, attn_mask=1):

FILE: tts/modules/ar_dur/commons/rot_transformer.py
  class SinusoidalPositionalEmbedding (line 29) | class SinusoidalPositionalEmbedding(nn.Module):
    method __init__ (line 35) | def __init__(self, embedding_dim, padding_idx, init_size=1024):
    method get_embedding (line 47) | def get_embedding(num_embeddings, embedding_dim, padding_idx=None):
    method forward (line 65) | def forward(self, input, incremental_state=None, timestep=None, positi...
    method max_positions (line 86) | def max_positions(self):
  class RotaryEmbeddings (line 91) | class RotaryEmbeddings(nn.Module):
    method __init__ (line 96) | def __init__(
    method _create_rotary_embed (line 134) | def _create_rotary_embed(self, *, width: int, length: int):
    method _rotate (line 149) | def _rotate(self, input: torch.Tensor):
    method forward (line 164) | def forward(self, input: torch.Tensor, *, positions: Optional[torch.Te...
  class RotMultiheadAttention (line 206) | class RotMultiheadAttention(MultiheadAttention):
    method __init__ (line 207) | def __init__(self, embed_dim, num_heads, kdim=None, vdim=None, dropout...
    method forward (line 215) | def forward(
  class RotMultiheadAttention2 (line 404) | class RotMultiheadAttention2(MultiheadAttention):
    method __init__ (line 405) | def __init__(self, embed_dim, num_heads, kdim=None, vdim=None, dropout...
    method forward (line 413) | def forward(
  class RotDecSALayer (line 543) | class RotDecSALayer(nn.Module):
    method __init__ (line 544) | def __init__(self, c, num_heads, dropout, attention_dropout=0.1, relu_...
    method forward (line 559) | def forward(
    method clear_buffer (line 604) | def clear_buffer(self, input, encoder_out=None, encoder_padding_mask=N...
    method set_buffer (line 608) | def set_buffer(self, name, tensor, incremental_state):
  class RotDecSALayer2 (line 612) | class RotDecSALayer2(RotDecSALayer):
    method __init__ (line 613) | def __init__(self, c, num_heads, dropout, attention_dropout=0.1, relu_...
  class RotTransformerDecoderLayer (line 622) | class RotTransformerDecoderLayer(nn.Module):
    method __init__ (line 623) | def __init__(self, hidden_size, dropout, kernel_size=9, num_heads=8, f...
    method forward (line 642) | def forward(self, x, **kwargs):
    method clear_buffer (line 645) | def clear_buffer(self, *args):
    method set_buffer (line 648) | def set_buffer(self, *args):

FILE: tts/modules/ar_dur/commons/seq_utils.py
  function make_positions (line 20) | def make_positions(tensor, padding_idx):
  function softmax (line 35) | def softmax(x, dim):
  function sequence_mask (line 39) | def sequence_mask(lengths, maxlen=None, dtype=torch.bool):
  function weights_nonzero_speech (line 47) | def weights_nonzero_speech(target):
  function _get_full_incremental_state_key (line 57) | def _get_full_incremental_state_key(module_instance, key):
  function get_incremental_state (line 69) | def get_incremental_state(module, incremental_state, key):
  function set_incremental_state (line 77) | def set_incremental_state(module, incremental_state, key, value):
  function fill_with_neg_inf (line 84) | def fill_with_neg_inf(t):
  function fill_with_neg_inf2 (line 89) | def fill_with_neg_inf2(t):
  function select_attn (line 94) | def select_attn(attn_logits, type='best'):
  function make_pad_mask (line 112) | def make_pad_mask(lengths, xs=None, length_dim=-1):
  function make_non_pad_mask (line 218) | def make_non_pad_mask(lengths, xs=None, length_dim=-1):
  function get_mask_from_lengths (line 298) | def get_mask_from_lengths(lengths):
  function group_hidden_by_segs (line 305) | def group_hidden_by_segs(h, seg_ids, max_len):
  function expand_by_repeat_times (line 321) | def expand_by_repeat_times(source_encoding, lengths):
  function expand_word2ph (line 338) | def expand_word2ph(word_encoding, ph2word):

FILE: tts/modules/ar_dur/commons/transformer.py
  class SinusoidalPositionalEmbedding (line 27) | class SinusoidalPositionalEmbedding(nn.Module):
    method __init__ (line 33) | def __init__(self, embedding_dim, padding_idx, init_size=1024):
    method get_embedding (line 45) | def get_embedding(num_embeddings, embedding_dim, padding_idx=None):
    method forward (line 63) | def forward(self, input, incremental_state=None, timestep=None, positi...
    method max_positions (line 84) | def max_positions(self):
  class TransformerFFNLayer (line 89) | class TransformerFFNLayer(nn.Module):
    method __init__ (line 90) | def __init__(self, hidden_size, filter_size, padding="SAME", kernel_si...
    method forward (line 105) | def forward(self, x, incremental_state=None):
    method _get_input_buffer (line 129) | def _get_input_buffer(self, incremental_state):
    method _set_input_buffer (line 136) | def _set_input_buffer(self, incremental_state, buffer):
    method clear_buffer (line 144) | def clear_buffer(self, incremental_state):
  class MultiheadAttention (line 152) | class MultiheadAttention(nn.Module):
    method __init__ (line 153) | def __init__(self, embed_dim, num_heads, kdim=None, vdim=None, dropout...
    method reset_parameters (line 201) | def reset_parameters(self):
    method forward (line 218) | def forward(
    method in_proj_qkv (line 432) | def in_proj_qkv(self, query):
    method in_proj_q (line 435) | def in_proj_q(self, query):
    method in_proj_k (line 444) | def in_proj_k(self, key):
    method in_proj_v (line 454) | def in_proj_v(self, value):
    method _in_proj (line 464) | def _in_proj(self, input, start=0, end=None):
    method _get_input_buffer (line 472) | def _get_input_buffer(self, incremental_state):
    method _set_input_buffer (line 479) | def _set_input_buffer(self, incremental_state, buffer):
    method apply_sparse_mask (line 487) | def apply_sparse_mask(self, attn_weights, tgt_len, src_len, bsz):
    method clear_buffer (line 490) | def clear_buffer(self, incremental_state=None):
  class EncSALayer (line 500) | class EncSALayer(nn.Module):
    method __init__ (line 501) | def __init__(self, c, num_heads, dropout, attention_dropout=0.1,
    method forward (line 516) | def forward(self, x, encoder_padding_mask=None, **kwargs):
  class DecSALayer (line 543) | class DecSALayer(nn.Module):
    method __init__ (line 544) | def __init__(self, c, num_heads, dropout, attention_dropout=0.1, relu_...
    method forward (line 562) | def forward(
    method clear_buffer (line 631) | def clear_buffer(self, input, encoder_out=None, encoder_padding_mask=N...
    method set_buffer (line 635) | def set_buffer(self, name, tensor, incremental_state):
  class TransformerEncoderLayer (line 639) | class TransformerEncoderLayer(nn.Module):
    method __init__ (line 640) | def __init__(self, hidden_size, dropout, kernel_size=9, num_heads=2, f...
    method forward (line 650) | def forward(self, x, **kwargs):
  class TransformerDecoderLayer (line 654) | class TransformerDecoderLayer(nn.Module):
    method __init__ (line 655) | def __init__(self, hidden_size, dropout, kernel_size=9, num_heads=2, f...
    method forward (line 666) | def forward(self, x, **kwargs):
    method clear_buffer (line 669) | def clear_buffer(self, *args):
    method set_buffer (line 672) | def set_buffer(self, *args):
  class FFTBlocks (line 676) | class FFTBlocks(nn.Module):
    method __init__ (line 677) | def __init__(self, hidden_size, num_layers, ffn_kernel_size=9, dropout...
    method forward (line 706) | def forward(self, x, padding_mask=None, attn_mask=None, return_hiddens...
  class FastSpeechEncoder (line 734) | class FastSpeechEncoder(FFTBlocks):
    method __init__ (line 735) | def __init__(self, dict_size, hidden_size=256, num_layers=4, kernel_si...
    method forward (line 746) | def forward(self, txt_tokens, attn_mask=None, other_embeds=0):
    method forward_embedding (line 760) | def forward_embedding(self, txt_tokens):

FILE: tts/modules/llm_dit/cfm.py
  class LogitNormalTrainingTimesteps (line 36) | class LogitNormalTrainingTimesteps:
    method __init__ (line 37) | def __init__(self, T=1000.0, loc=0.0, scale=1.0):
    method sample (line 42) | def sample(self, size, device):
  function pad_t_like_x (line 47) | def pad_t_like_x(t, x):
  class ConditionalFlowMatcher (line 71) | class ConditionalFlowMatcher:
    method __init__ (line 82) | def __init__(self, sigma: Union[float, int] = 0.0):
    method compute_mu_t (line 92) | def compute_mu_t(self, x0, x1, t):
    method compute_sigma_t (line 115) | def compute_sigma_t(self, t):
    method sample_xt (line 134) | def sample_xt(self, x0, x1, t, epsilon):
    method compute_conditional_flow (line 161) | def compute_conditional_flow(self, x0, x1, t, xt):
    method sample_noise_like (line 186) | def sample_noise_like(self, x):
    method sample_location_and_conditional_flow (line 189) | def sample_location_and_conditional_flow(self, x0, x1, t=None, return_...
    method compute_lambda (line 233) | def compute_lambda(self, t):
  class VariancePreservingConditionalFlowMatcher (line 252) | class VariancePreservingConditionalFlowMatcher(ConditionalFlowMatcher):
    method compute_mu_t (line 260) | def compute_mu_t(self, x0, x1, t):
    method compute_conditional_flow (line 282) | def compute_conditional_flow(self, x0, x1, t, xt):

FILE: tts/modules/llm_dit/dit.py
  class Diffusion (line 27) | class Diffusion(nn.Module):
    method __init__ (line 28) | def __init__(self):
    method forward (line 70) | def forward(self, inputs, sigmas=None, x_noisy=None):
    method forward_ling_encoder (line 101) | def forward_ling_encoder(self, txt_tokens, tone_tokens):
    method _forward (line 114) | def _forward(self, x, local_cond, x_ling, timesteps, ctx_mask, dur=Non...
    method inference (line 127) | def inference(self, inputs, timesteps=20, seq_cfg_w=[1.0, 1.0], **kwar...

FILE: tts/modules/llm_dit/time_embedding.py
  class SinusPositionEmbedding (line 20) | class SinusPositionEmbedding(nn.Module):
    method __init__ (line 21) | def __init__(self, dim):
    method forward (line 25) | def forward(self, x, scale=1000):
  class TimestepEmbedding (line 34) | class TimestepEmbedding(nn.Module):
    method __init__ (line 35) | def __init__(self, dim, freq_embed_dim=256):
    method forward (line 40) | def forward(self, timestep):  # noqa: F821

FILE: tts/modules/llm_dit/transformer.py
  function precompute_freqs_cis (line 23) | def precompute_freqs_cis(dim: int, end: int, theta: float = 10000.0):
  function reshape_for_broadcast (line 31) | def reshape_for_broadcast(freqs_cis: torch.Tensor, x: torch.Tensor):
  function apply_rotary_emb (line 39) | def apply_rotary_emb(
  class AdaLNZero (line 52) | class AdaLNZero(nn.Module):
    method __init__ (line 53) | def __init__(self, dim):
    method forward (line 59) | def forward(self, x, emb=None):
  class AdaLNZero_Out (line 66) | class AdaLNZero_Out(nn.Module):
    method __init__ (line 67) | def __init__(self, dim):
    method forward (line 73) | def forward(self, x, emb):
  class Attention (line 80) | class Attention(nn.Module):
    method __init__ (line 81) | def __init__(self, encoder_dim, encoder_n_heads, max_seq_len):
    method forward (line 107) | def forward(
  class FeedForward (line 130) | class FeedForward(nn.Module):
    method __init__ (line 131) | def __init__(
    method forward (line 150) | def forward(self, x):
  class TransformerBlock (line 154) | class TransformerBlock(nn.Module):
    method __init__ (line 155) | def __init__(self, encoder_dim, encoder_n_heads, max_seq_len):
    method forward (line 170) | def forward(
  class Transformer (line 207) | class Transformer(nn.Module):
    method __init__ (line 208) | def __init__(self, encoder_n_layers, encoder_dim, encoder_n_heads, max...
    method forward (line 224) | def forward(self, x, t, attn_mask, start_pos=0):

FILE: tts/modules/wavvae/decoder/diag_gaussian.py
  class DiagonalGaussianDistribution (line 18) | class DiagonalGaussianDistribution(object):
    method __init__ (line 19) | def __init__(self, parameters: torch.Tensor, deterministic: bool = Fal...
    method sample (line 31) | def sample(self, generator=None) -> torch.Tensor:
    method kl (line 42) | def kl(self, other: "DiagonalGaussianDistribution" = None) -> torch.Te...
    method nll (line 57) | def nll(self, sample, dims) -> torch.Tensor:
    method mode (line 66) | def mode(self) -> torch.Tensor:

FILE: tts/modules/wavvae/decoder/hifigan_modules.py
  function init_weights (line 25) | def init_weights(m, mean=0.0, std=0.01):
  function get_padding (line 31) | def get_padding(kernel_size, dilation=1):
  class Upsample (line 35) | class Upsample(nn.Module):
    method __init__ (line 36) | def __init__(self, mult, r):
    method forward (line 52) | def forward(self, x):
  class Downsample (line 59) | class Downsample(nn.Module):
    method __init__ (line 60) | def __init__(self, mult, r):
    method forward (line 70) | def forward(self, x):
  function weights_init (line 75) | def weights_init(m):
  function weights_zero_init (line 84) | def weights_zero_init(m):
  function WNConv1d (line 91) | def WNConv1d(*args, **kwargs):
  function WNConvTranspose1d (line 95) | def WNConvTranspose1d(*args, **kwargs):
  class Audio2Mel (line 99) | class Audio2Mel(nn.Module):
    method __init__ (line 100) | def __init__(
    method forward (line 129) | def forward(self, audio):
  class ResnetBlock (line 149) | class ResnetBlock(nn.Module):
    method __init__ (line 150) | def __init__(self, dim, dilation=1, dim_in=None):
    method forward (line 164) | def forward(self, x):
  class ResBlockMRFV2 (line 174) | class ResBlockMRFV2(torch.nn.Module):
    method __init__ (line 175) | def __init__(self, channels, kernel_size=3, dilation=(1, 3, 5)):
    method forward (line 197) | def forward(self, x):
    method remove_weight_norm (line 206) | def remove_weight_norm(self):
  class ResBlockMRFV2Inter (line 213) | class ResBlockMRFV2Inter(torch.nn.Module):
    method __init__ (line 214) | def __init__(self, channels, kernel_size=3):
    method forward (line 220) | def forward(self, x):
  class Generator (line 228) | class Generator(nn.Module):
    method __init__ (line 229) | def __init__(self, input_size_, ngf, n_residual_layers, num_band, args...
    method forward (line 265) | def forward(self, mel, step=None):

FILE: tts/modules/wavvae/decoder/seanet_encoder.py
  class Encoder (line 21) | class Encoder(nn.Module):
    method __init__ (line 22) | def __init__(
    method forward (line 35) | def forward(self, audio: torch.Tensor):

FILE: tts/modules/wavvae/decoder/wavvae_v3.py
  class WavVAE_V3 (line 25) | class WavVAE_V3(nn.Module):
    method __init__ (line 26) | def __init__(self, hparams=None):
    method encode_latent (line 41) | def encode_latent(self, audio):
    method encode (line 46) | def encode(self, audio):
    method decode (line 52) | def decode(self, latent):
    method forward (line 56) | def forward(self, audio):

FILE: tts/modules/wavvae/encoder/common_modules/conv.py
  function apply_parametrization_norm (line 47) | def apply_parametrization_norm(module: nn.Module, norm: str = 'none') ->...
  function get_norm_module (line 57) | def get_norm_module(module: nn.Module, causal: bool = False, norm: str =...
  function get_extra_padding_for_conv1d (line 71) | def get_extra_padding_for_conv1d(x: torch.Tensor, kernel_size: int, stri...
  function pad1d (line 79) | def pad1d(x: torch.Tensor, paddings: tp.Tuple[int, int], mode: str = 'ze...
  class ConvLayerNorm (line 96) | class ConvLayerNorm(nn.LayerNorm):
    method __init__ (line 97) | def __init__(self, normalized_shape: tp.Union[int, tp.List[int], torch...
    method forward (line 100) | def forward(self, x):
  class NormConv1d (line 107) | class NormConv1d(nn.Module):
    method __init__ (line 108) | def __init__(self, *args, causal: bool = False, norm: str = 'none',
    method forward (line 115) | def forward(self, x):
  class SConv1d (line 121) | class SConv1d(nn.Module):
    method __init__ (line 122) | def __init__(self, in_channels: int, out_channels: int,
    method forward (line 138) | def forward(self, x):

FILE: tts/modules/wavvae/encoder/common_modules/lstm.py
  class SLSTM (line 34) | class SLSTM(nn.Module):
    method __init__ (line 39) | def __init__(self, dimension: int, num_layers: int = 2, skip: bool = T...
    method forward (line 45) | def forward(self, x):

FILE: tts/modules/wavvae/encoder/common_modules/seanet.py
  class SEANetResnetBlock (line 41) | class SEANetResnetBlock(nn.Module):
    method __init__ (line 42) | def __init__(self, dim: int, kernel_sizes: tp.List[int] = [3, 1], dila...
    method forward (line 68) | def forward(self, x):
  class SEANetEncoder (line 72) | class SEANetEncoder(nn.Module):
    method __init__ (line 73) | def __init__(self, channels: int = 1, dimension: int = 128, n_filters:...
    method forward (line 125) | def forward(self, x):

FILE: tts/utils/audio_utils/align.py
  function mel2token_to_dur (line 17) | def mel2token_to_dur(mel2token, T_txt=None, max_dur=None):

FILE: tts/utils/audio_utils/io.py
  function to_wav_bytes (line 25) | def to_wav_bytes(wav, sr, norm=False):
  function save_wav (line 39) | def save_wav(wav_bytes, path):
  function to_mp3 (line 46) | def to_mp3(out_path):
  function convert_to_wav (line 55) | def convert_to_wav(wav_path):
  function convert_to_wav_bytes (line 73) | def convert_to_wav_bytes(audio_binary):
  function combine_audio_segments (line 83) | def combine_audio_segments(segments, crossfade_duration=0.16, sr=24000):

FILE: tts/utils/audio_utils/plot.py
  function spec_to_figure (line 25) | def spec_to_figure(spec, vmin=None, vmax=None, title='', f0s=None, dur_i...
  function align_to_figure (line 73) | def align_to_figure(align, dur_info):

FILE: tts/utils/commons/ckpt_utils.py
  function dist_load (line 28) | def dist_load(path):
  function torch_load_dist (line 48) | def torch_load_dist(path, map_location='cpu'):
  function get_last_checkpoint (line 54) | def get_last_checkpoint(work_dir, steps=None):
  function get_all_ckpts (line 64) | def get_all_ckpts(work_dir, steps=None):
  function load_ckpt (line 73) | def load_ckpt(cur_model, ckpt_base_dir, model_name='model', force=True, ...
  function load_with_size_mismatch (line 161) | def load_with_size_mismatch(model, state_dict, prefix=""):

FILE: tts/utils/commons/hparams.py
  class Args (line 26) | class Args:
    method __init__ (line 27) | def __init__(self, **kwargs):
  function override_config (line 32) | def override_config(old_config: dict, new_config: dict):
  function traverse_dict (line 42) | def traverse_dict(d, func, ctx):
  function parse_config (line 51) | def parse_config(v, context=None):
  function remove_meta_key (line 66) | def remove_meta_key(d):
  function load_config (line 76) | def load_config(config_fn, config_chains, loaded_configs):
  function set_hparams (line 103) | def set_hparams(config='', exp_name='', hparams_str='', print_hparams=Tr...

FILE: tts/utils/text_utils/ph_tone_convert.py
  function map_phone_to_tokendict (line 18) | def map_phone_to_tokendict(item, pad_bos_eos=True):
  function split_ph_timestamp (line 39) | def split_ph_timestamp(ph_timestamp):
  function split_ph (line 72) | def split_ph(ph_seq):

FILE: tts/utils/text_utils/split_text.py
  function chunk_text_chinese (line 18) | def chunk_text_chinese(text, limit=60):
  function chunk_text_english (line 60) | def chunk_text_english(text, max_chars=130):
  function chunk_text_chinesev2 (line 90) | def chunk_text_chinesev2(text, limit=60, look_ahead_limit=30):

FILE: tts/utils/text_utils/text_encoder.py
  function strip_ids (line 44) | def strip_ids(ids, ids_to_strip):
  class TextEncoder (line 52) | class TextEncoder(object):
    method __init__ (line 55) | def __init__(self, num_reserved_ids=NUM_RESERVED_TOKENS):
    method num_reserved_ids (line 59) | def num_reserved_ids(self):
    method encode (line 62) | def encode(self, s):
    method decode (line 78) | def decode(self, ids, strip_extraneous=False):
    method decode_list (line 95) | def decode_list(self, ids):
    method vocab_size (line 117) | def vocab_size(self):
  class TokenTextEncoder (line 121) | class TokenTextEncoder(TextEncoder):
    method __init__ (line 124) | def __init__(self,
    method encode (line 161) | def encode(self, s):
    method decode (line 174) | def decode(self, ids, strip_eos=False, strip_padding=False):
    method decode_list (line 183) | def decode_list(self, ids):
    method vocab_size (line 188) | def vocab_size(self):
    method __len__ (line 191) | def __len__(self):
    method _safe_id_to_token (line 194) | def _safe_id_to_token(self, idx):
    method _init_vocab_from_file (line 197) | def _init_vocab_from_file(self, filename):
    method _init_vocab_from_list (line 212) | def _init_vocab_from_list(self, vocab_list):
    method _init_vocab (line 229) | def _init_vocab(self, token_generator, add_reserved_tokens=True):
    method pad (line 245) | def pad(self):
    method eos (line 248) | def eos(self):
    method unk (line 251) | def unk(self):
    method seg (line 254) | def seg(self):
    method store_to_file (line 257) | def store_to_file(self, filename):
    method sil_phonemes (line 270) | def sil_phonemes(self):
  function build_token_encoder (line 274) | def build_token_encoder(token_list_file):
  function is_sil_phoneme (line 279) | def is_sil_phoneme(p):