MCPcopy Create free account

hub / github.com/FireRedTeam/FireRedTTS / functions

Functions342 in github.com/FireRedTeam/FireRedTTS

↓ 15 callersMethod__init__
( self, dimension: int, num_layers: int = 2, bidirectional: bool = False,
fireredtts/modules/semantic_tokenizer/semantic_tokenizer.py:337
↓ 12 callersMethod__init__
( self, in_channels: int, out_channels: int, kernel_size: int, stride:
fireredtts/modules/acoustic_codec/bigcodec.py:152
↓ 9 callersMethod__init__
( self, out_channels, kernel_size, input_shape=None, in_channels=None,
fireredtts/modules/semantic_tokenizer/ecapa_tdnn.py:124
↓ 8 callersMethod__init__
( self, in_features:int, hidden_features:Optional[int]=None, o
fireredtts/modules/flowmatching/estimator_dit.py:9
↓ 8 callersMethod__init__
(self, channels: int, out_channels: int, stride: int = 2)
fireredtts/modules/flowmatching/upsample_encoder.py:287
↓ 8 callersFunctionget_padding
(kernel_size, dilation=1)
fireredtts/modules/bigvgan/bigvgan.py:17
↓ 7 callersFunctionWNConv1d
(*args, **kwargs)
fireredtts/modules/acoustic_codec/bigcodec.py:270
↓ 5 callersFunctionutf_8_len
(text)
fireredtts/utils/spliter.py:74
↓ 4 callersFunctionapply_parametrization_norm
(module: nn.Module, norm: str = "none")
fireredtts/modules/semantic_tokenizer/semantic_tokenizer.py:58
↓ 4 callersFunctionget_norm_module
( module: nn.Module, causal: bool = False, norm: str = "none", **norm_kwargs )
fireredtts/modules/semantic_tokenizer/semantic_tokenizer.py:70
↓ 4 callersFunctionmodulate
(x, shift, scale)
fireredtts/modules/flowmatching/estimator_dit.py:108
↓ 3 callersMethod__init__
Args:
fireredtts/modules/semantic_llm/llm_gpt2.py:357
↓ 3 callersFunctionbreak_text
(texts, length, splits: set)
fireredtts/utils/spliter.py:78
↓ 3 callersFunctionclean_text
(text)
fireredtts/utils/spliter.py:57
↓ 3 callersMethoddecode
(self, embed_ind)
fireredtts/modules/acoustic_codec/vector_quantization.py:161
↓ 3 callersMethoddequantize
(self, embed_ind)
fireredtts/modules/acoustic_codec/vector_quantization.py:146
↓ 3 callersMethoddownsample
(self, x, x_len)
fireredtts/modules/semantic_tokenizer/semantic_tokenizer.py:521
↓ 3 callersMethodencode
(self, x)
fireredtts/modules/acoustic_codec/vector_quantization.py:150
↓ 3 callersMethodgenerate
(self, cond_latents, text_inputs, **hf_generate_kwargs)
fireredtts/modules/semantic_llm/llm_gpt2.py:513
↓ 3 callersMethodto_heads
(self, ts:torch.Tensor)
fireredtts/modules/flowmatching/estimator_dit.py:69
↓ 3 callersMethodupsample
(self, x, x_len)
fireredtts/modules/semantic_tokenizer/semantic_tokenizer.py:528
↓ 2 callersMethod__init__
( self, # Model n_stacks=2, layers=12, model_dim=1024, heads=1
fireredtts/modules/acoustic_llm/acoustic_llm.py:536
↓ 2 callersMethod__init__
( self, dim: int, codebook_size: int, decay: float = 0.99, epsilon: fl
fireredtts/modules/acoustic_codec/vector_quantization.py:51
↓ 2 callersMethod__init__
( self, num_mels: int, upsample_initial_channel: int, resblock_kernel_sizes: t
fireredtts/modules/bigvgan/bigvgan.py:237
↓ 2 callersMethod_extract
(self, inp, length)
fireredtts/modules/semantic_tokenizer/hubert.py:61
↓ 2 callersFunctionadd_cleaned
(curr, segments)
fireredtts/utils/spliter.py:114
↓ 2 callersFunctionapply_parametrization_norm
(module: nn.Module, norm: str = "none")
fireredtts/modules/acoustic_codec/bigcodec.py:37
↓ 2 callersMethodcompute_embeddings
(self, cond_latents, text_inputs)
fireredtts/modules/semantic_llm/llm_gpt2.py:494
↓ 2 callersFunctioncontains_chinese
(text)
fireredtts/modules/text_normalizer/utils.py:88
↓ 2 callersFunctionema_inplace
(moving_avg, new, decay)
fireredtts/modules/acoustic_codec/vector_quantization.py:21
↓ 2 callersMethodextend_pe
Reset the positional encodings.
fireredtts/modules/flowmatching/upsample_encoder.py:198
↓ 2 callersFunctionget_extra_padding_for_conv1d
( x: torch.Tensor, kernel_size: int, stride: int, padding_total: int = 0 )
fireredtts/modules/acoustic_codec/bigcodec.py:60
↓ 2 callersFunctionget_extra_padding_for_conv1d
( x: torch.Tensor, kernel_size: int, stride: int, padding_total: int = 0 )
fireredtts/modules/semantic_tokenizer/semantic_tokenizer.py:86
↓ 2 callersMethodget_fixed_embedding
(self, ind, dev)
fireredtts/modules/semantic_llm/llm_gpt2.py:308
↓ 2 callersFunctionget_norm_module
( module: nn.Module, causal: bool = False, norm: str = "none", **norm_kwargs )
fireredtts/modules/acoustic_codec/bigcodec.py:47
↓ 2 callersFunctionis_chinese
(char)
fireredtts/modules/text_normalizer/utils.py:96
↓ 2 callersFunctionkaiser_sinc_filter1d
( cutoff, half_width, kernel_size )
fireredtts/modules/acoustic_codec/alias_free_torch/filter.py:30
↓ 2 callersFunctionkaiser_sinc_filter1d
( cutoff, half_width, kernel_size )
fireredtts/modules/bigvgan/alias_free_torch/filter.py:29
↓ 2 callersFunctionlength_to_mask
Creates a binary mask for each sequence. Reference: https://discuss.pytorch.org/t/how-to-generate-variable-length-mask/23397/3 Arguments
fireredtts/modules/semantic_tokenizer/ecapa_tdnn.py:14
↓ 2 callersFunctionload_audio
_summary_ Args: audiopath (_type_): audio_path sampling_rate (_type_): sampling_rate Returns: _type_: _description_
fireredtts/utils/utils.py:9
↓ 2 callersMethodload_model
(self, acoustic_llm_path, acoustic_codec_path)
fireredtts/models/token2audio.py:15
↓ 2 callersFunctionpad1d
( x: torch.Tensor, paddings: tp.Tuple[int, int], mode: str = "zero", value: float = 0.0, )
fireredtts/modules/acoustic_codec/bigcodec.py:76
↓ 2 callersFunctionpad1d
( x: torch.Tensor, paddings: tp.Tuple[int, int], mode: str = "zero", value: float = 0.0, )
fireredtts/modules/semantic_tokenizer/semantic_tokenizer.py:102
↓ 2 callersMethodpostprocess_emb
(self, embed_ind, shape)
fireredtts/modules/acoustic_codec/vector_quantization.py:143
↓ 2 callersMethodpreprocess
(self, x)
fireredtts/modules/acoustic_codec/vector_quantization.py:128
↓ 2 callersMethodquantize
(self, x)
fireredtts/modules/acoustic_codec/vector_quantization.py:132
↓ 2 callersMethodremove_weight_norm
(self)
fireredtts/modules/acoustic_codec/bigcodec.py:495
↓ 2 callersMethodstore_prefix_emb
(self, prefix_emb)
fireredtts/modules/semantic_llm/llm_gpt2.py:25
↓ 2 callersMethodsynthesize_base
_summary_ Args: prompt_wav (_type_): _description_ prompt_text (_type_): _description_ text (_type_): _de
fireredtts/models/fireredtts.py:112
↓ 2 callersMethodtn
(self, text)
fireredtts/modules/text_normalizer/normalize.py:122
↓ 2 callersFunctionunpad1d
(x: torch.Tensor, paddings: tp.Tuple[int, int])
fireredtts/modules/acoustic_codec/bigcodec.py:98
↓ 2 callersFunctionunpad1d
(x: torch.Tensor, paddings: tp.Tuple[int, int])
fireredtts/modules/semantic_tokenizer/semantic_tokenizer.py:124
↓ 1 callersFunctionWNConvTranspose1d
(*args, **kwargs)
fireredtts/modules/acoustic_codec/bigcodec.py:280
↓ 1 callersMethod__init__
(self, ratio=2, kernel_size=None)
fireredtts/modules/acoustic_codec/alias_free_torch/resample.py:11
↓ 1 callersMethod__init__
Initialization. INPUT: - in_features: shape of the input - alpha: trainable parameter alpha is in
fireredtts/modules/bigvgan/activations.py:26
↓ 1 callersMethod__init__
(self, ratio=2, kernel_size=None)
fireredtts/modules/bigvgan/alias_free_torch/resample.py:10
↓ 1 callersMethod__init__
( self, channels:int=512, )
fireredtts/modules/flowmatching/flow.py:10
↓ 1 callersMethod__init__
(self, config)
fireredtts/models/token2audio.py:10
↓ 1 callersMethod_check_input_shape
Checks the input shape and returns the number of input channels.
fireredtts/modules/semantic_tokenizer/ecapa_tdnn.py:237
↓ 1 callersMethod_dropout
(self, quants, enabled=True)
fireredtts/modules/acoustic_codec/vector_quantization.py:419
↓ 1 callersMethod_euler
( self, x0: torch.Tensor, c: torch.Tensor, n_timesteps: int = 10, )
fireredtts/modules/flowmatching/flow.py:68
↓ 1 callersMethod_forward
(self, x, g=None)
fireredtts/modules/semantic_tokenizer/semantic_tokenizer.py:494
↓ 1 callersMethod_integrate
(self, quants, losses, indices, update_codebook=True)
fireredtts/modules/acoustic_codec/vector_quantization.py:372
↓ 1 callersMethod_manage_padding
This function performs zero-padding on the time axis such that their lengths is unchanged after the convolution. Arguments --
fireredtts/modules/semantic_tokenizer/ecapa_tdnn.py:204
↓ 1 callersMethod_postprocess
(self, feats, lengths, normalize=False)
fireredtts/modules/semantic_tokenizer/hubert.py:98
↓ 1 callersMethod_preprocess
(self, x)
fireredtts/modules/acoustic_codec/vector_quantization.py:337
↓ 1 callersMethod_quantize
(self, x, x_len, enable_vq, update_codebook)
fireredtts/modules/acoustic_codec/vector_quantization.py:342
↓ 1 callersFunctionbreak_text_by_length
(texts, length)
fireredtts/utils/spliter.py:96
↓ 1 callersMethodbuild_aligned_inputs_and_targets
( self, seqs, lens, start_token, stop_token, delay=0 )
fireredtts/modules/acoustic_llm/acoustic_llm.py:676
↓ 1 callersFunctionbuild_hf_gpt_transformer
( layers, model_dim, heads, max_mel_seq_len, max_text_seq_len, checkpointing )
fireredtts/modules/acoustic_llm/acoustic_llm.py:507
↓ 1 callersFunctionbuild_hf_gpt_transformer
GPT-2 implemented by the HuggingFace library.
fireredtts/modules/semantic_llm/llm_gpt2.py:312
↓ 1 callersMethodcompute_embeddings_ic
_summary_ Args: cond_latents (_type_): speaker embedding text_inputs (_type_): text tokens prompt_tokens
fireredtts/modules/semantic_llm/llm_gpt2.py:530
↓ 1 callersMethoddecode
(self, seqs, seq_lens=None)
fireredtts/modules/semantic_tokenizer/semantic_tokenizer.py:629
↓ 1 callersFunctiondefault
(val: tp.Any, d: tp.Any)
fireredtts/modules/acoustic_codec/vector_quantization.py:12
↓ 1 callersMethoddeserialize
(self, seqs, seq_lens)
fireredtts/modules/semantic_tokenizer/semantic_tokenizer.py:668
↓ 1 callersFunctiondynamic_range_compression_torch
(x, C=1, clip_val=1e-5)
fireredtts/modules/bigvgan/mel_spectrogram.py:11
↓ 1 callersFunctiondynamic_range_compression_torch
(x, C=1, clip_val=1e-5)
fireredtts/modules/semantic_tokenizer/audio.py:19
↓ 1 callersFunctiondynamic_range_decompression_torch
(x, C=1)
fireredtts/modules/semantic_tokenizer/audio.py:23
↓ 1 callersMethodexpire_codes_
(self, batch_samples, dists=None)
fireredtts/modules/acoustic_codec/vector_quantization.py:110
↓ 1 callersMethodextract
(self, wavs, wav_lengths)
fireredtts/modules/semantic_tokenizer/__init__.py:29
↓ 1 callersMethodextract
(self, wavs, wav_lengths, spk)
fireredtts/models/token2audio.py:48
↓ 1 callersMethodextract_speech_tokens
( self, wav, wav_length, serialize=True, extract_spk=True, shuffle=False )
fireredtts/modules/acoustic_codec/bigcodec.py:667
↓ 1 callersMethodextract_speech_tokens
( self, wav, wav_length, serialize=True, extract_spk=True, shuffle=False )
fireredtts/modules/semantic_tokenizer/semantic_tokenizer.py:820
↓ 1 callersMethodextract_spk_embeddings
(self, prompt_wav)
fireredtts/models/fireredtts.py:85
↓ 1 callersFunctionf2b
(ustr, exemption="。,:")
fireredtts/modules/text_normalizer/utils.py:116
↓ 1 callersMethodforward
( self, wav, wav_length=None, enable_vq=True, decode=True, update_codebook=True )
fireredtts/modules/acoustic_codec/bigcodec.py:620
↓ 1 callersMethodforward
(self, x)
fireredtts/modules/flowmatching/estimator_dit.py:29
↓ 1 callersMethodforward
Args: tokens: shape (b, t) Returns: token_embs: shape (b, t, c)
fireredtts/modules/flowmatching/flow.py:23
↓ 1 callersMethodforward
(self, x)
fireredtts/modules/semantic_tokenizer/semantic_tokenizer.py:354
↓ 1 callersMethodforward
( self, wav, wav_length, enable_vq=True, decode=True, extract_
fireredtts/modules/semantic_tokenizer/semantic_tokenizer.py:754
↓ 1 callersMethodforward_attention
Compute attention context vector. Args: value (torch.Tensor): shape: (b, nh, t2, c//nh). scores (torch.Tensor): shape:
fireredtts/modules/flowmatching/upsample_encoder.py:46
↓ 1 callersMethodforward_qkv
Args: query,key,value: shape (b, t, c) Returns: query,key,value: shape (b, nh, t, c//nh)
fireredtts/modules/flowmatching/upsample_encoder.py:27
↓ 1 callersMethodgenerate_ic
_summary_ Args: cond_latents (_type_): _description_ text_inputs (_type_): _description_ prompt_tokens (_
fireredtts/modules/semantic_llm/llm_gpt2.py:568
↓ 1 callersMethodget_conditioning
(self, speech_conditioning_input)
fireredtts/modules/acoustic_llm/acoustic_llm.py:753
↓ 1 callersMethodget_fixed_embedding
(self, ind, dev)
fireredtts/modules/acoustic_llm/acoustic_llm.py:503
↓ 1 callersFunctionget_mask_from_lengths
(lengths, max_len=None)
fireredtts/modules/semantic_tokenizer/hubert.py:10
↓ 1 callersFunctionget_padding_elem
This function computes the number of elements to add for zero-padding. Arguments --------- L_in : int stride: int kernel_size : i
fireredtts/modules/semantic_tokenizer/ecapa_tdnn.py:62
↓ 1 callersMethodinference
( self, prompt_token: torch.Tensor, prompt_xvec: torch.Tensor, prompt_feat: to
fireredtts/modules/flowmatching/flow.py:102
↓ 1 callersMethodinference_speech
( self, speech_conditioning_latent, text_inputs, input_tokens=None, nu
fireredtts/modules/acoustic_llm/acoustic_llm.py:763
↓ 1 callersMethodinit_embed_
(self, data)
fireredtts/modules/acoustic_codec/vector_quantization.py:85
next →1–100 of 342, ranked by callers