library_name: transformers
tags:
- 文本转语音
- 标注
license: apache-2.0
language:
- 英文
pipeline_tag: text-to-speech
inference: false
datasets:
- parler-tts/mls_eng_10k
- blabble-io/libritts_r
- parler-tts/libritts_r_tags_tagged_10k_generated
- parler-tts/mls-eng-10k-tags_tagged_10k_generated
Parler-TTS Mini v0.1
Parler-TTS Mini v0.1 是一个轻量级的文本转语音(TTS)模型,基于10.5K小时的音频数据训练而成,能够生成高质量、自然流畅的语音,并通过简单的文本提示控制语音特征(如性别、背景噪音、语速、音调和混响效果)。这是 Parler-TTS 项目的首个发布模型,该项目旨在为社区提供TTS训练资源和数据集预处理代码。
使用方法
使用Parler-TTS就像说“你好”一样简单。只需安装一次库:
pip install git+https://github.com/huggingface/parler-tts.git
然后可以使用以下推理代码片段运行模型:
import torch
from parler_tts import ParlerTTSForConditionalGeneration
from transformers import AutoTokenizer
import soundfile as sf
device = "cuda:0" if torch.cuda.is_available() else "cpu"
model = ParlerTTSForConditionalGeneration.from_pretrained("parler-tts/parler_tts_mini_v0.1").to(device)
tokenizer = AutoTokenizer.from_pretrained("parler-tts/parler_tts_mini_v0.1")
prompt = "嘿,你今天过得怎么样?"
description = "一位音调略低的女声非常富有表现力地说话,声音来自一个听起来非常封闭的环境,音频质量清晰。她说话非常快。"
input_ids = tokenizer(description, return_tensors="pt").input_ids.to(device)
prompt_input_ids = tokenizer(prompt, return_tensors="pt").input_ids.to(device)
generation = model.generate(input_ids=input_ids, prompt_input_ids=prompt_input_ids)
audio_arr = generation.cpu().numpy().squeeze()
sf.write("parler_tts_out.wav", audio_arr, model.config.sampling_rate)
提示:
- 包含“非常清晰的音频”可以生成最高质量的音频,而“非常嘈杂的音频”则会生成高背景噪音的音频
- 可以使用标点符号控制生成的韵律,例如使用逗号在语音中添加短暂停顿
- 其他语音特征(性别、语速、音调和混响效果)可以直接通过提示控制
项目背景
Parler-TTS是对Stability AI和爱丁堡大学的Dan Lyth与Simon King的论文《Natural language guidance of high-fidelity text-to-speech with synthetic annotations》工作的复现。
与其他TTS模型不同,Parler-TTS是一个完全开源的发布版本。所有数据集、预处理、训练代码和权重均在宽松许可下公开,使社区能够基于我们的工作开发自己的强大TTS模型。Parler-TTS的发布包括:
引用
如果您觉得这个仓库有用,请考虑引用我们的工作以及原始的Stability AI论文:
@misc{lacombe-etal-2024-parler-tts,
author = {Yoach Lacombe and Vaibhav Srivastav and Sanchit Gandhi},
title = {Parler-TTS},
year = {2024},
publisher = {GitHub},
journal = {GitHub repository},
howpublished = {\url{https://github.com/huggingface/parler-tts}}
}
@misc{lyth2024natural,
title={Natural language guidance of high-fidelity text-to-speech with synthetic annotations},
author={Dan Lyth and Simon King},
year={2024},
eprint={2402.01912},
archivePrefix={arXiv},
primaryClass={cs.SD}
}
许可
本模型采用Apache 2.0许可协议授权。