Skip to main content

espnet2.gan_svs.vits.generator.VISingerGenerator

About 5 min

espnet2.gan_svs.vits.generator.VISingerGenerator

source

class espnet2.gan_svs.vits.generator.VISingerGenerator(vocabs: int, aux_channels: int = 513, hidden_channels: int = 192, spks: int | None = None, langs: int | None = None, spk_embed_dim: int | None = None, global_channels: int = -1, segment_size: int = 32, text_encoder_attention_heads: int = 2, text_encoder_ffn_expand: int = 4, text_encoder_blocks: int = 6, text_encoder_positionwise_layer_type: str = 'conv1d', text_encoder_positionwise_conv_kernel_size: int = 1, text_encoder_positional_encoding_layer_type: str = 'rel_pos', text_encoder_self_attention_layer_type: str = 'rel_selfattn', text_encoder_activation_type: str = 'swish', text_encoder_normalize_before: bool = True, text_encoder_dropout_rate: float = 0.1, text_encoder_positional_dropout_rate: float = 0.0, text_encoder_attention_dropout_rate: float = 0.0, text_encoder_conformer_kernel_size: int = 7, use_macaron_style_in_text_encoder: bool = True, use_conformer_conv_in_text_encoder: bool = True, decoder_kernel_size: int = 7, decoder_channels: int = 512, decoder_downsample_scales: List[int] = [2, 2, 8, 8], decoder_downsample_kernel_sizes: List[int] = [4, 4, 16, 16], decoder_upsample_scales: List[int] = [8, 8, 2, 2], decoder_upsample_kernel_sizes: List[int] = [16, 16, 4, 4], decoder_resblock_kernel_sizes: List[int] = [3, 7, 11], decoder_resblock_dilations: List[List[int]] = [[1, 3, 5], [1, 3, 5], [1, 3, 5]], use_avocodo=False, projection_filters: List[int] = [0, 1, 1, 1], projection_kernels: List[int] = [0, 5, 7, 11], n_harmonic: int = 64, use_weight_norm_in_decoder: bool = True, posterior_encoder_kernel_size: int = 5, posterior_encoder_layers: int = 16, posterior_encoder_stacks: int = 1, posterior_encoder_base_dilation: int = 1, posterior_encoder_dropout_rate: float = 0.0, use_weight_norm_in_posterior_encoder: bool = True, flow_flows: int = 4, flow_kernel_size: int = 5, flow_base_dilation: int = 1, flow_layers: int = 4, flow_dropout_rate: float = 0.0, use_weight_norm_in_flow: bool = True, use_only_mean_in_flow: bool = True, generator_type: str = 'visinger', vocoder_generator_type: str = 'hifigan', fs: int = 22050, hop_length: int = 256, win_length: int | None = 1024, n_fft: int = 1024, use_phoneme_predictor: bool = False, expand_f0_method: str = 'repeat', hubert_channels: int | None = 0)

Bases: Module

Generator module in VISinger.

Initialize VITS generator module.

  • Parameters:
    • vocabs (int) – Input vocabulary size.
    • aux_channels (int) – Number of acoustic feature channels.
    • hidden_channels (int) – Number of hidden channels.
    • spks (Optional *[*int ]) – Number of speakers. If set to > 1, assume that the sids will be provided as the input and use sid embedding layer.
    • langs (Optional *[*int ]) – Number of languages. If set to > 1, assume that the lids will be provided as the input and use sid embedding layer.
    • spk_embed_dim (Optional *[*int ]) – Speaker embedding dimension. If set to > 0, assume that spembs will be provided as the input.
    • global_channels (int) – Number of global conditioning channels.
    • segment_size (int) – Segment size for decoder.
    • text_encoder_attention_heads (int) – Number of heads in conformer block of text encoder.
    • text_encoder_ffn_expand (int) – Expansion ratio of FFN in conformer block of text encoder.
    • text_encoder_blocks (int) – Number of conformer blocks in text encoder.
    • text_encoder_positionwise_layer_type (str) – Position-wise layer type in conformer block of text encoder.
    • text_encoder_positionwise_conv_kernel_size (int) – Position-wise convolution kernel size in conformer block of text encoder. Only used when the above layer type is conv1d or conv1d-linear.
    • text_encoder_positional_encoding_layer_type (str) – Positional encoding layer type in conformer block of text encoder.
    • text_encoder_self_attention_layer_type (str) – Self-attention layer type in conformer block of text encoder.
    • text_encoder_activation_type (str) – Activation function type in conformer block of text encoder.
    • text_encoder_normalize_before (bool) – Whether to apply layer norm before self-attention in conformer block of text encoder.
    • text_encoder_dropout_rate (float) – Dropout rate in conformer block of text encoder.
    • text_encoder_positional_dropout_rate (float) – Dropout rate for positional encoding in conformer block of text encoder.
    • text_encoder_attention_dropout_rate (float) – Dropout rate for attention in conformer block of text encoder.
    • text_encoder_conformer_kernel_size (int) – Conformer conv kernel size. It will be used when only use_conformer_conv_in_text_encoder = True.
    • use_macaron_style_in_text_encoder (bool) – Whether to use macaron style FFN in conformer block of text encoder.
    • use_conformer_conv_in_text_encoder (bool) – Whether to use covolution in conformer block of text encoder.
    • decoder_kernel_size (int) – Decoder kernel size.
    • decoder_channels (int) – Number of decoder initial channels.
    • decoder_downsample_scales (List *[*int ]) – List of downsampling scales in decoder.
    • decoder_downsample_kernel_sizes (List *[*int ]) – List of kernel sizes for downsampling layers in decoder.
    • decoder_upsample_scales (List *[*int ]) – List of upsampling scales in decoder.
    • decoder_upsample_kernel_sizes (List *[*int ]) – List of kernel sizes for upsampling layers in decoder.
    • decoder_resblock_kernel_sizes (List *[*int ]) – List of kernel sizes for resblocks in decoder.
    • decoder_resblock_dilations (List *[*List *[*int ] ]) – List of list of dilations for resblocks in decoder.
    • use_avocodo (bool) – Whether to use Avocodo model in the generator.
    • projection_filters (List *[*int ]) – List of projection filter sizes.
    • projection_kernels (List *[*int ]) – List of projection kernel sizes.
    • n_harmonic (int) – Number of harmonic components.
    • use_weight_norm_in_decoder (bool) – Whether to apply weight normalization in decoder.
    • posterior_encoder_kernel_size (int) – Posterior encoder kernel size.
    • posterior_encoder_layers (int) – Number of layers of posterior encoder.
    • posterior_encoder_stacks (int) – Number of stacks of posterior encoder.
    • posterior_encoder_base_dilation (int) – Base dilation of posterior encoder.
    • posterior_encoder_dropout_rate (float) – Dropout rate for posterior encoder.
    • use_weight_norm_in_posterior_encoder (bool) – Whether to apply weight normalization in posterior encoder.
    • flow_flows (int) – Number of flows in flow.
    • flow_kernel_size (int) – Kernel size in flow.
    • flow_base_dilation (int) – Base dilation in flow.
    • flow_layers (int) – Number of layers in flow.
    • flow_dropout_rate (float) – Dropout rate in flow
    • use_weight_norm_in_flow (bool) – Whether to apply weight normalization in flow.
    • use_only_mean_in_flow (bool) – Whether to use only mean in flow.
    • generator_type (str) – Type of generator to use for the model.
    • vocoder_generator_type (str) – Type of vocoder generator to use for the model.
    • fs (int) – Sample rate of the audio.
    • hop_length (int) – Number of samples between successive frames in STFT.
    • win_length (int) – Window size of the STFT.
    • n_fft (int) – Length of the FFT window to be used.
    • use_phoneme_predictor (bool) – Whether to use phoneme predictor in the model.
    • expand_f0_method (str) – The method used to expand F0. Use “repeat” or “interpolation”.
    • hubert_channels (int) – Number of channels in the Hubert model. This is used in VISinger2 Plus.

forward(text: Tensor, text_lengths: Tensor, feats: Tensor, feats_lengths: Tensor, label: Tensor | None = None, label_lengths: Tensor | None = None, melody: Tensor | None = None, gt_dur: Tensor | None = None, score_dur: Tensor | None = None, slur: Tensor | None = None, pitch: Tensor | None = None, ying: Tensor | None = None, sids: Tensor | None = None, spembs: Tensor | None = None, lids: Tensor | None = None) → Tuple[Tensor, Tensor, Tensor, Tensor, Tensor, Tensor, Tuple[Tensor, Tensor, Tensor, Tensor, Tensor, Tensor]]

Calculate forward propagation.

  • Parameters:
    • text (LongTensor) – Batch of padded character ids (B, Tmax).
    • text_lengths (LongTensor) – Batch of lengths of each input batch (B,).
    • feats (Tensor) – Batch of padded target features (B, Lmax, odim).
    • feats_lengths (LongTensor) – Batch of the lengths of each target (B,).
    • label (LongTensor) – Batch of padded label ids (B, Tmax).
    • label_lengths (LongTensor) – Batch of the lengths of padded label ids (B, ).
    • melody (LongTensor) – Batch of padded midi (B, Tmax).
    • gt_dur (LongTensor) – Batch of padded ground truth duration (B, Tmax).
    • score_dur (LongTensor) – Batch of padded score duration (B, Tmax).
    • pitch (FloatTensor) – Batch of padded f0 (B, Tmax).
    • ying (Optional *[*Tensor ]) – Batch of padded ying (B, Tmax).
    • spembs (Optional *[*Tensor ]) – Batch of speaker embeddings (B, spk_embed_dim).
    • sids (Optional *[*Tensor ]) – Batch of speaker IDs (B, 1).
    • lids (Optional *[*Tensor ]) – Batch of language IDs (B, 1).
  • Returns: Waveform tensor (B, 1, segment_size * upsample_factor). Tensor: Duration negative log-likelihood (NLL) tensor (B,). Tensor: Monotonic attention weight tensor (B, 1, T_feats, T_text). Tensor: Segments start index tensor (B,). Tensor: Text mask tensor (B, 1, T_text). Tensor: Feature mask tensor (B, 1, T_feats). tuple[Tensor, Tensor, Tensor, Tensor, Tensor, Tensor]:
    • Tensor: Posterior encoder hidden representation (B, H, T_feats).
    • Tensor: Flow hidden representation (B, H, T_feats).
    • Tensor: Expanded text encoder projected mean (B, H, T_feats).
    • Tensor: Expanded text encoder projected scale (B, H, T_feats).
    • Tensor: Posterior encoder projected mean (B, H, T_feats).
    • Tensor: Posterior encoder projected scale (B, H, T_feats).
  • Return type: Tensor

inference(text: Tensor, text_lengths: Tensor, feats: Tensor | None = None, feats_lengths: Tensor | None = None, label: Tensor | None = None, label_lengths: Tensor | None = None, melody: Tensor | None = None, score_dur: Tensor | None = None, slur: Tensor | None = None, gt_dur: Tensor | None = None, pitch: Tensor | None = None, sids: Tensor | None = None, spembs: Tensor | None = None, lids: Tensor | None = None, noise_scale: float = 0.667, noise_scale_dur: float = 0.8, alpha: float = 1.0, max_len: int | None = None, use_teacher_forcing: bool = False) → Tuple[Tensor, Tensor, Tensor]

Run inference.

  • Parameters:
    • text (LongTensor) – Batch of padded character ids (B, Tmax).
    • text_lengths (LongTensor) – Batch of lengths of each input batch (B,).
    • feats (Tensor) – Batch of padded target features (B, Lmax, odim).
    • feats_lengths (LongTensor) – Batch of the lengths of each target (B,).
    • label (LongTensor) – Batch of padded label ids (B, Tmax).
    • label_lengths (LongTensor) – Batch of the lengths of padded label ids (B, ).
    • melody (LongTensor) – Batch of padded midi (B, Tmax).
    • gt_dur (LongTensor) – Batch of padded ground truth duration (B, Tmax).
    • score_dur (LongTensor) – Batch of padded score duration (B, Tmax).
    • pitch (FloatTensor) – Batch of padded f0 (B, Tmax).
    • ying (Optional *[*Tensor ]) – Batch of padded ying (B, Tmax).
    • spembs (Optional *[*Tensor ]) – Batch of speaker embeddings (B, spk_embed_dim).
    • sids (Optional *[*Tensor ]) – Batch of speaker IDs (B, 1).
    • lids (Optional *[*Tensor ]) – Batch of language IDs (B, 1).
    • noise_scale (float) – Noise scale parameter for flow.
    • noise_scale_dur (float) – Noise scale parameter for duration predictor.
    • alpha (float) – Alpha parameter to control the speed of generated speech.
    • max_len (Optional *[*int ]) – Maximum length of acoustic feature sequence.
    • use_teacher_forcing (bool) – Whether to use teacher forcing.
  • Returns: Generated waveform tensor (B, T_wav).
  • Return type: Tensor
Copyright © 2024 ESPnet Community. All rights reserved.