Transformers 文档
VITS
并获取增强的文档体验
开始使用
VITS
概述
VITS 模型由 Jaehyeon Kim、Jungil Kong、Juhee Son 在 Conditional Variational Autoencoder with Adversarial Learning for End-to-End Text-to-Speech 中提出。
VITS (Variational Inference with adversarial learning for end-to-end Text-to-Speech) 是一种端到端语音合成模型,用于预测以输入文本序列为条件的语音波形。它是一种条件变分自动编码器 (VAE),由后验编码器、解码器和条件先验组成。
一组基于频谱图的声学特征由基于流的模块预测,该模块由基于 Transformer 的文本编码器和多个耦合层组成。频谱图使用一系列转置卷积层进行解码,风格与 HiFi-GAN 声码器非常相似。受 TTS 问题的一对多性质(即相同的文本输入可以用多种方式说出)的启发,该模型还包括一个随机时长预测器,该预测器允许模型从相同的输入文本合成具有不同节奏的语音。
该模型通过变分下界和对抗训练中获得的损失组合进行端到端训练。为了提高模型的表达能力,将归一化流应用于条件先验分布。在推理过程中,文本编码根据时长预测模块进行上采样,然后使用流模块和 HiFi-GAN 解码器的级联映射到波形中。由于时长预测器的随机性,该模型是非确定性的,因此需要固定种子才能生成相同的语音波形。
该论文的摘要如下
最近提出了几种支持单阶段训练和平行采样的端到端文本到语音 (TTS) 模型,但它们的样本质量与两阶段 TTS 系统不匹配。在这项工作中,我们提出了一种并行端到端 TTS 方法,该方法生成比当前两阶段模型更自然的音频。我们的方法采用了增强归一化流的变分推理和对抗训练过程,这提高了生成建模的表达能力。我们还提出了一个随机时长预测器,用于从输入文本合成具有不同节奏的语音。通过潜变量的不确定性建模和随机时长预测器,我们的方法表达了自然的一对多关系,其中文本输入可以用不同的音调和节奏以多种方式说出。在 LJ Speech(一个单说话人数据集)上的主观人工评估(平均意见得分或 MOS)表明,我们的方法优于最佳公开可用的 TTS 系统,并实现了与真实值相当的 MOS。
该模型还可以与来自 Massively Multilingual Speech (MMS) 的 TTS 检查点一起使用,因为这些检查点使用相同的架构和略微修改的分词器。
此模型由 Matthijs 和 sanchit-gandhi 贡献。原始代码可以在这里找到。
使用示例
VITS 和 MMS-TTS 检查点都可以使用相同的 API。由于基于流的模型是非确定性的,因此最好设置种子以确保输出的可重复性。对于使用罗马字母的语言,例如英语或法语,可以直接使用分词器来预处理文本输入。以下代码示例使用 MMS-TTS 英语检查点运行正向传递
import torch
from transformers import VitsTokenizer, VitsModel, set_seed
tokenizer = VitsTokenizer.from_pretrained("facebook/mms-tts-eng")
model = VitsModel.from_pretrained("facebook/mms-tts-eng")
inputs = tokenizer(text="Hello - my dog is cute", return_tensors="pt")
set_seed(555) # make deterministic
with torch.no_grad():
outputs = model(**inputs)
waveform = outputs.waveform[0]
生成的波形可以保存为 .wav
文件
import scipy
scipy.io.wavfile.write("techno.wav", rate=model.config.sampling_rate, data=waveform)
或在 Jupyter Notebook / Google Colab 中显示
from IPython.display import Audio
Audio(waveform, rate=model.config.sampling_rate)
对于某些使用非罗马字母的语言,例如阿拉伯语、普通话或印地语,需要 uroman
perl 包来预处理为罗马字母的文本输入。
您可以通过检查预训练的 tokenizer
的 is_uroman
属性来检查您的语言是否需要 uroman
包
from transformers import VitsTokenizer
tokenizer = VitsTokenizer.from_pretrained("facebook/mms-tts-eng")
print(tokenizer.is_uroman)
如果 is_uroman 属性为 True
,则分词器将自动将 uroman
包应用于您的文本输入,但您需要安装 uroman(如果尚未安装),使用
pip install --upgrade uroman
注意:使用 uroman
作为 python 包所需的 Python 版本应 >= 3.10
。您可以像往常一样使用分词器,无需任何额外的预处理步骤
import torch
from transformers import VitsTokenizer, VitsModel, set_seed
import os
import subprocess
tokenizer = VitsTokenizer.from_pretrained("facebook/mms-tts-kor")
model = VitsModel.from_pretrained("facebook/mms-tts-kor")
text = "이봐 무슨 일이야"
inputs = tokenizer(text=text, return_tensors="pt")
set_seed(555) # make deterministic
with torch.no_grad():
outputs = model(inputs["input_ids"])
waveform = outputs.waveform[0]
如果您不想升级到 python >= 3.10
,那么您可以使用 uroman
perl 包来预处理为罗马字母的文本输入。为此,首先将 uroman 存储库克隆到您的本地计算机,并将 bash 变量 UROMAN
设置为本地路径
git clone https://github.com/isi-nlp/uroman.git
cd uroman
export UROMAN=$(pwd)
然后,您可以使用以下代码片段预处理文本输入。您可以依赖于使用 bash 变量 UROMAN
指向 uroman 存储库,也可以将 uroman 目录作为参数传递给 uromanize
函数
import torch
from transformers import VitsTokenizer, VitsModel, set_seed
import os
import subprocess
tokenizer = VitsTokenizer.from_pretrained("facebook/mms-tts-kor")
model = VitsModel.from_pretrained("facebook/mms-tts-kor")
def uromanize(input_string, uroman_path):
"""Convert non-Roman strings to Roman using the `uroman` perl package."""
script_path = os.path.join(uroman_path, "bin", "uroman.pl")
command = ["perl", script_path]
process = subprocess.Popen(command, stdin=subprocess.PIPE, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
# Execute the perl command
stdout, stderr = process.communicate(input=input_string.encode())
if process.returncode != 0:
raise ValueError(f"Error {process.returncode}: {stderr.decode()}")
# Return the output as a string and skip the new-line character at the end
return stdout.decode()[:-1]
text = "이봐 무슨 일이야"
uromanized_text = uromanize(text, uroman_path=os.environ["UROMAN"])
inputs = tokenizer(text=uromanized_text, return_tensors="pt")
set_seed(555) # make deterministic
with torch.no_grad():
outputs = model(inputs["input_ids"])
waveform = outputs.waveform[0]
VitsConfig
class transformers.VitsConfig
< source >( vocab_size = 38 hidden_size = 192 num_hidden_layers = 6 num_attention_heads = 2 window_size = 4 use_bias = True ffn_dim = 768 layerdrop = 0.1 ffn_kernel_size = 3 flow_size = 192 spectrogram_bins = 513 hidden_act = 'relu' hidden_dropout = 0.1 attention_dropout = 0.1 activation_dropout = 0.1 initializer_range = 0.02 layer_norm_eps = 1e-05 use_stochastic_duration_prediction = True num_speakers = 1 speaker_embedding_size = 0 upsample_initial_channel = 512 upsample_rates = [8, 8, 2, 2] upsample_kernel_sizes = [16, 16, 4, 4] resblock_kernel_sizes = [3, 7, 11] resblock_dilation_sizes = [[1, 3, 5], [1, 3, 5], [1, 3, 5]] leaky_relu_slope = 0.1 depth_separable_channels = 2 depth_separable_num_layers = 3 duration_predictor_flow_bins = 10 duration_predictor_tail_bound = 5.0 duration_predictor_kernel_size = 3 duration_predictor_dropout = 0.5 duration_predictor_num_flows = 4 duration_predictor_filter_channels = 256 prior_encoder_num_flows = 4 prior_encoder_num_wavenet_layers = 4 posterior_encoder_num_wavenet_layers = 16 wavenet_kernel_size = 5 wavenet_dilation_rate = 1 wavenet_dropout = 0.0 speaking_rate = 1.0 noise_scale = 0.667 noise_scale_duration = 0.8 sampling_rate = 16000 **kwargs )
参数
- vocab_size (
int
, 可选, 默认为 38) — VITS 模型的词汇表大小。定义了传递给 VitsModel 前向方法的inputs_ids
可以表示的不同标记的数量。 - hidden_size (
int
, 可选, 默认为 192) — 文本编码器层的维度。 - num_hidden_layers (
int
, 可选, 默认为 6) — Transformer 编码器中隐藏层的数量。 - num_attention_heads (
int
, 可选, 默认为 2) — Transformer 编码器中每个注意力层的注意力头数。 - window_size (
int
, 可选, 默认为 4) — Transformer 编码器中注意力层中相对位置嵌入的窗口大小。 - use_bias (
bool
, 可选, 默认为True
) — 是否在 Transformer 编码器中的键、查询、值投影层中使用偏置。 - ffn_dim (
int
, optional, defaults to 768) — Transformer 编码器中“中间”(即,前馈)层的维度。 - layerdrop (
float
, optional, defaults to 0.1) — 编码器的 LayerDrop 概率。 参阅 [LayerDrop 论文](see https://arxiv.org/abs/1909.11556) 了解更多详情。 - ffn_kernel_size (
int
, optional, defaults to 3) — Transformer 编码器中前馈网络使用的 1D 卷积层的内核大小。 - flow_size (
int
, optional, defaults to 192) — 流动层的维度。 - spectrogram_bins (
int
, optional, defaults to 513) — 目标频谱图中的频率箱数量。 - hidden_act (
str
或function
, optional, defaults to"relu"
) — 编码器和池化器中的非线性激活函数(函数或字符串)。如果为字符串,则支持"gelu"
、"relu"
、"selu"
和"gelu_new"
。 - hidden_dropout (
float
, optional, defaults to 0.1) — 嵌入层和编码器中所有全连接层的 dropout 概率。 - attention_dropout (
float
, optional, defaults to 0.1) — 注意力概率的 dropout 比率。 - activation_dropout (
float
, optional, defaults to 0.1) — 全连接层内部激活的 dropout 比率。 - initializer_range (
float
, optional, defaults to 0.02) — 用于初始化所有权重矩阵的 truncated_normal_initializer 的标准差。 - layer_norm_eps (
float
, optional, defaults to 1e-05) — 层归一化层使用的 epsilon 值。 - use_stochastic_duration_prediction (
bool
, optional, defaults toTrue
) — 是否使用随机时长预测模块或常规时长预测器。 - num_speakers (
int
, optional, defaults to 1) — 如果是多说话人模型,则为说话人数量。 - speaker_embedding_size (
int
, optional, defaults to 0) — 说话人嵌入使用的通道数。 对于单说话人模型为零。 - upsample_initial_channel (
int
, optional, defaults to 512) — HiFi-GAN 上采样网络的输入通道数。 - upsample_rates (
Tuple[int]
或List[int]
, optional, defaults to[8, 8, 2, 2]
) — 定义 HiFi-GAN 上采样网络中每个 1D 卷积层步幅的整数元组。upsample_rates
的长度定义了卷积层的数量,并且必须与upsample_kernel_sizes
的长度匹配。 - upsample_kernel_sizes (
Tuple[int]
或List[int]
, optional, defaults to[16, 16, 4, 4]
) — 定义 HiFi-GAN 上采样网络中每个 1D 卷积层内核大小的整数元组。upsample_kernel_sizes
的长度定义了卷积层的数量,并且必须与upsample_rates
的长度匹配。 - resblock_kernel_sizes (
Tuple[int]
或List[int]
, optional, defaults to[3, 7, 11]
) — 定义 HiFi-GAN 多感受野融合 (MRF) 模块中 1D 卷积层内核大小的整数元组。 - resblock_dilation_sizes (
Tuple[Tuple[int]]
或List[List[int]]
, optional, defaults to[[1, 3, 5], [1, 3, 5], [1, 3, 5]]
) — 定义 HiFi-GAN 多感受野融合 (MRF) 模块中扩张 1D 卷积层扩张率的嵌套整数元组。 - leaky_relu_slope (
float
, optional, defaults to 0.1) — leaky ReLU 激活使用的负斜率角度。 - depth_separable_channels (
int
, optional, defaults to 2) — 每个深度可分离块中要使用的通道数。 - depth_separable_num_layers (
int
, optional, defaults to 3) — 每个深度可分离块中要使用的卷积层数。 - duration_predictor_flow_bins (
int
, optional, defaults to 10) — 使用时长预测模型中无约束有理样条映射的通道数。 - duration_predictor_tail_bound (
float
, optional, defaults to 5.0) — 在时长预测模型中计算无约束有理样条时,尾部箱体边界的值。 - duration_predictor_kernel_size (
int
, optional, defaults to 3) — 时长预测模型中使用的 1D 卷积层的内核大小。 - duration_predictor_dropout (
float
, optional, defaults to 0.5) — 时长预测模型的 dropout 比率。 - duration_predictor_num_flows (
int
, optional, defaults to 4) — 时长预测模型使用的流动阶段数。 - duration_predictor_filter_channels (
int
, optional, defaults to 256) — 用于时长预测模型中卷积层的通道数。 - prior_encoder_num_flows (
int
, optional, defaults to 4) — 先验编码器流动模型使用的流动阶段数。 - prior_encoder_num_wavenet_layers (
int
, optional, defaults to 4) — 先验编码器流动模型使用的 WaveNet 层数。 - posterior_encoder_num_wavenet_layers (
int
, optional, defaults to 16) — 后验编码器模型使用的 WaveNet 层数。 - wavenet_kernel_size (
int
, 可选, 默认为 5) — WaveNet 模型中使用的 1D 卷积层的内核大小。 - wavenet_dilation_rate (
int
, 可选, 默认为 1) — WaveNet 模型中使用的空洞 1D 卷积层的空洞率。 - wavenet_dropout (
float
, 可选, 默认为 0.0) — WaveNet 层的 dropout 比率。 - speaking_rate (
float
, 可选, 默认为 1.0) — 语速。值越大,合成语音速度越快。 - noise_scale (
float
, 可选, 默认为 0.667) — 语音预测的随机程度。值越大,预测语音的变化越大。 - noise_scale_duration (
float
, 可选, 默认为 0.8) — 时长预测的随机程度。值越大,预测时长的变化越大。 - sampling_rate (
int
, 可选, 默认为 16000) — 输出音频波形数字化的采样率,以赫兹 (Hz) 表示。
这是用于存储 VitsModel 配置的配置类。 它用于根据指定的参数实例化 VITS 模型,定义模型架构。 使用默认值实例化配置将产生类似于 VITS facebook/mms-tts-eng 架构的配置。
配置对象继承自 PretrainedConfig,可用于控制模型输出。 有关更多信息,请阅读 PretrainedConfig 中的文档。
示例
>>> from transformers import VitsModel, VitsConfig
>>> # Initializing a "facebook/mms-tts-eng" style configuration
>>> configuration = VitsConfig()
>>> # Initializing a model (with random weights) from the "facebook/mms-tts-eng" style configuration
>>> model = VitsModel(configuration)
>>> # Accessing the model configuration
>>> configuration = model.config
VitsTokenizer
class transformers.VitsTokenizer
< source >( vocab_file pad_token = '<pad>' unk_token = '<unk>' language = None add_blank = True normalize = True phonemize = True is_uroman = False **kwargs )
构建 VITS tokenizer。 还支持 MMS-TTS。
此 tokenizer 继承自 PreTrainedTokenizer,其中包含大多数主要方法。 用户应参考此超类以获取有关这些方法的更多信息。
__call__
< source >( text: typing.Union[str, typing.List[str], typing.List[typing.List[str]]] = None text_pair: typing.Union[str, typing.List[str], typing.List[typing.List[str]], NoneType] = None text_target: typing.Union[str, typing.List[str], typing.List[typing.List[str]]] = None text_pair_target: typing.Union[str, typing.List[str], typing.List[typing.List[str]], NoneType] = None add_special_tokens: bool = True padding: typing.Union[bool, str, transformers.utils.generic.PaddingStrategy] = False truncation: typing.Union[bool, str, transformers.tokenization_utils_base.TruncationStrategy] = None max_length: typing.Optional[int] = None stride: int = 0 is_split_into_words: bool = False pad_to_multiple_of: typing.Optional[int] = None padding_side: typing.Optional[str] = None return_tensors: typing.Union[str, transformers.utils.generic.TensorType, NoneType] = None return_token_type_ids: typing.Optional[bool] = None return_attention_mask: typing.Optional[bool] = None return_overflowing_tokens: bool = False return_special_tokens_mask: bool = False return_offsets_mapping: bool = False return_length: bool = False verbose: bool = True **kwargs ) → BatchEncoding
参数
- text (
str
,List[str]
,List[List[str]]
, 可选) — 要编码的序列或序列批次。 每个序列可以是字符串或字符串列表(预 token 化的字符串)。 如果序列以字符串列表(预 token 化)形式提供,则必须设置is_split_into_words=True
(以消除与序列批次的歧义)。 - text_pair (
str
,List[str]
,List[List[str]]
, 可选) — 要编码的序列或序列批次。 每个序列可以是字符串或字符串列表(预 token 化的字符串)。 如果序列以字符串列表(预 token 化)形式提供,则必须设置is_split_into_words=True
(以消除与序列批次的歧义)。 - text_target (
str
,List[str]
,List[List[str]]
, 可选) — 要编码为目标文本的序列或序列批次。 每个序列可以是字符串或字符串列表(预 token 化的字符串)。 如果序列以字符串列表(预 token 化)形式提供,则必须设置is_split_into_words=True
(以消除与序列批次的歧义)。 - text_pair_target (
str
,List[str]
,List[List[str]]
, 可选) — 要编码为目标文本的序列或序列批次。 每个序列可以是字符串或字符串列表(预 token 化的字符串)。 如果序列以字符串列表(预 token 化)形式提供,则必须设置is_split_into_words=True
(以消除与序列批次的歧义)。 - add_special_tokens (
bool
, 可选, 默认为True
) — 在编码序列时是否添加特殊 token。 这将使用底层的PretrainedTokenizerBase.build_inputs_with_special_tokens
函数,该函数定义了哪些 token 会自动添加到 input ids 中。 如果您想自动添加bos
或eos
token,这将非常有用。 - padding (
bool
,str
或 PaddingStrategy, 可选, 默认为False
) — 激活并控制填充。 接受以下值:True
或'longest'
:填充到批次中最长的序列(如果仅提供单个序列,则不填充)。'max_length'
:填充到由参数max_length
指定的最大长度,如果未提供该参数,则填充到模型可接受的最大输入长度。False
或'do_not_pad'
(默认):不填充(即,可以输出具有不同长度序列的批次)。
- truncation (
bool
,str
或 TruncationStrategy, 可选, 默认为False
) — 激活并控制截断。 接受以下值:True
或'longest_first'
:截断为由参数max_length
指定的最大长度,如果未提供该参数,则截断为模型可接受的最大输入长度。 如果提供了一对序列(或一批对),这将逐个 token 进行截断,从对中最长的序列中删除 token。'only_first'
:截断为由参数max_length
指定的最大长度,如果未提供该参数,则截断为模型可接受的最大输入长度。 如果提供了一对序列(或一批对),这将仅截断对中的第一个序列。'only_second'
:截断为由参数max_length
指定的最大长度,如果未提供该参数,则截断为模型可接受的最大输入长度。 如果提供了一对序列(或一批对),这将仅截断对中的第二个序列。False
或'do_not_truncate'
(默认):不截断(即,可以输出序列长度大于模型最大允许输入大小的批次)。
- max_length (
int
, 可选) — 控制截断/填充参数之一使用的最大长度。如果保留未设置或设置为
None
,如果截断/填充参数之一需要最大长度,则将使用预定义的模型最大长度。 如果模型没有特定的最大输入长度(如 XLNet),则将停用截断/填充到最大长度。 - stride (
int
, 可选, 默认为 0) — 如果与max_length
一起设置为数字,则当return_overflowing_tokens=True
时返回的溢出 token 将包含来自截断序列末尾的一些 token,以在截断序列和溢出序列之间提供一些重叠。 此参数的值定义了重叠 token 的数量。 - is_split_into_words (
bool
, optional, defaults toFalse
) — 输入是否已预先分词(例如,分成单词)。如果设置为True
,分词器会假定输入已经分成单词(例如,通过空格分割),然后对其进行分词。这对于 NER 或 token 分类非常有用。 - pad_to_multiple_of (
int
, optional) — 如果设置,将序列填充到提供值的倍数。需要激活padding
。这对于在计算能力>= 7.5
(Volta) 的 NVIDIA 硬件上启用 Tensor Cores 的使用尤其有用。 - padding_side (
str
, optional) — 模型应该在其上应用填充的一侧。应从 [‘right’, ‘left’] 中选择。默认值从同名的类属性中选取。 - return_tensors (
str
或 TensorType, optional) — 如果设置,将返回张量而不是 Python 整数列表。可接受的值为:'tf'
: 返回 TensorFlowtf.constant
对象。'pt'
: 返回 PyTorchtorch.Tensor
对象。'np'
: 返回 Numpynp.ndarray
对象。
- return_token_type_ids (
bool
, optional) — 是否返回 token 类型 ID。如果保留为默认值,将根据特定分词器的默认值返回 token 类型 ID,由return_outputs
属性定义。 - return_attention_mask (
bool
, optional) — 是否返回 attention mask。如果保留为默认值,将根据特定分词器的默认值返回 attention mask,由return_outputs
属性定义。 - return_overflowing_tokens (
bool
, optional, defaults toFalse
) — 是否返回溢出的 token 序列。如果提供一对输入 id 序列(或一批序列对)且truncation_strategy = longest_first
或True
,则会引发错误,而不是返回溢出的 token。 - return_special_tokens_mask (
bool
, optional, defaults toFalse
) — 是否返回特殊 token mask 信息。 - return_offsets_mapping (
bool
, optional, defaults toFalse
) — 是否返回每个 token 的(char_start, char_end)
。这仅在继承自 PreTrainedTokenizerFast 的快速分词器上可用,如果使用 Python 的分词器,此方法将引发
NotImplementedError
。 - return_length (
bool
, optional, defaults toFalse
) — 是否返回编码输入的长度。 - verbose (
bool
, optional, defaults toTrue
) — 是否打印更多信息和警告。 - **kwargs — 传递给
self.tokenize()
方法
返回值
一个 BatchEncoding 对象,包含以下字段
-
input_ids — 要馈送到模型的 token ID 列表。
-
token_type_ids — 要馈送到模型的 token 类型 ID 列表(当
return_token_type_ids=True
或如果 “token_type_ids” 在self.model_input_names
中时)。 -
attention_mask — 指定模型应关注哪些 token 的索引列表(当
return_attention_mask=True
或如果 “attention_mask” 在self.model_input_names
中时)。 -
overflowing_tokens — 溢出的 token 序列列表(当指定了
max_length
且return_overflowing_tokens=True
时)。 -
num_truncated_tokens — 截断的 token 数量(当指定了
max_length
且return_overflowing_tokens=True
时)。 -
special_tokens_mask — 0 和 1 的列表,其中 1 指定添加的特殊 token,0 指定常规序列 token(当
add_special_tokens=True
且return_special_tokens_mask=True
时)。 -
length — 输入的长度(当
return_length=True
时)
用于分词并将一个或多个序列或一对或多对序列准备用于模型的主要方法。
VitsModel
class transformers.VitsModel
< source >( config: VitsConfig )
参数
- config (VitsConfig) — 包含模型所有参数的模型配置类。使用配置文件初始化不会加载与模型关联的权重,仅加载配置。查看 from_pretrained() 方法以加载模型权重。
完整的 VITS 模型,用于文本到语音合成。此模型继承自 PreTrainedModel。查看超类文档以了解库为所有模型实现的通用方法(例如下载或保存、调整输入嵌入大小、剪枝头等)。
此模型也是 PyTorch torch.nn.Module 子类。将其用作常规 PyTorch 模块,并参阅 PyTorch 文档以了解与一般用法和行为相关的所有事项。
forward
< source >( input_ids: typing.Optional[torch.Tensor] = None attention_mask: typing.Optional[torch.Tensor] = None speaker_id: typing.Optional[int] = None output_attentions: typing.Optional[bool] = None output_hidden_states: typing.Optional[bool] = None return_dict: typing.Optional[bool] = None labels: typing.Optional[torch.FloatTensor] = None ) → transformers.models.vits.modeling_vits.VitsModelOutput
或 tuple(torch.FloatTensor)
参数
- input_ids (
torch.LongTensor
,形状为(batch_size, sequence_length)
) — 词汇表中输入序列 token 的索引。如果您提供填充,默认情况下将被忽略。索引可以使用 AutoTokenizer 获取。有关详细信息,请参阅 PreTrainedTokenizer.encode() 和 PreTrainedTokenizer.call()。
- attention_mask (
torch.Tensor
,形状为(batch_size, sequence_length)
, optional) — Mask,用于避免在填充 token 索引上执行卷积和注意力。Mask 值在[0, 1]
中选择:- 1 表示 **未被 mask** 的 token,
- 0 表示 **已被 mask** 的 token。
- speaker_id (
int
, optional) — 要使用哪个说话人嵌入。仅用于多说话人模型。 - output_attentions (
bool
, optional) — 是否返回所有注意力层的注意力张量。有关更多详细信息,请参阅返回张量下的attentions
。 - output_hidden_states (
bool
, optional) — 是否返回所有层的隐藏状态。有关更多详细信息,请参阅返回张量下的hidden_states
。 - return_dict (
bool
, optional) — 是否返回 ModelOutput 而不是纯粹的元组 (plain tuple)。 - labels (
torch.FloatTensor
, 形状为(batch_size, config.spectrogram_bins, sequence_length)
, optional) — 目标频谱图的浮点数值。时间步设置为-100.0
的将被忽略(masked)用于损失计算。
返回值
transformers.models.vits.modeling_vits.VitsModelOutput
或 tuple(torch.FloatTensor)
一个 transformers.models.vits.modeling_vits.VitsModelOutput
或一个 torch.FloatTensor
元组 (如果传递了 return_dict=False
或者当 config.return_dict=False
时),包含各种元素,具体取决于配置 (VitsConfig) 和输入。
-
waveform (
torch.FloatTensor
, 形状为(batch_size, sequence_length)
) — 模型预测的最终音频波形。 -
sequence_lengths (
torch.FloatTensor
, 形状为(batch_size,)
) —waveform
批次中每个元素的样本长度。 -
spectrogram (
torch.FloatTensor
, 形状为(batch_size, sequence_length, num_bins)
) — 在流模型输出端预测的 log-mel 频谱图。此频谱图被传递到 Hi-Fi GAN 解码器模型以获得最终的音频波形。 -
hidden_states (
tuple(torch.FloatTensor)
, optional, 当传递了output_hidden_states=True
或当config.output_hidden_states=True
时返回) —torch.FloatTensor
元组(如果模型具有嵌入层,则为嵌入输出提供一个,+ 每个层的输出提供一个),形状为(batch_size, sequence_length, hidden_size)
。模型在每一层输出端的隐藏状态,加上可选的初始嵌入输出。
-
attentions (
tuple(torch.FloatTensor)
, optional, 当传递了output_attentions=True
或当config.output_attentions=True
时返回) —torch.FloatTensor
元组(每一层一个),形状为(batch_size, num_heads, sequence_length, sequence_length)
。注意力 softmax 之后的注意力权重,用于计算自注意力头中的加权平均值。
VitsModel 的 forward 方法,覆盖了 __call__
特殊方法。
虽然 forward 传递的配方需要在该函数中定义,但应该在之后调用 Module
实例而不是此函数,因为前者负责运行预处理和后处理步骤,而后者会默默地忽略它们。
示例
>>> from transformers import VitsTokenizer, VitsModel, set_seed
>>> import torch
>>> tokenizer = VitsTokenizer.from_pretrained("facebook/mms-tts-eng")
>>> model = VitsModel.from_pretrained("facebook/mms-tts-eng")
>>> inputs = tokenizer(text="Hello - my dog is cute", return_tensors="pt")
>>> set_seed(555) # make deterministic
>>> with torch.no_grad():
... outputs = model(inputs["input_ids"])
>>> outputs.waveform.shape
torch.Size([1, 45824])