库名称: transformers
标签:
- 文本转语音
- 标注
许可证: apache-2.0
支持语言:
- 英语
- 法语
- 西班牙语
- 葡萄牙语
- 波兰语
- 德语
- 荷兰语
- 意大利语
管道标签: 文本转语音
推理: false
数据集:
- facebook/multilingual_librispeech
- parler-tts/libritts_r_filtered
- parler-tts/libritts-r-filtered-speaker-descriptions
- parler-tts/mls_eng
- parler-tts/mls-eng-speaker-descriptions
- PHBJT/mls-annotated
- PHBJT/cml-tts-filtered-annotated
- PHBJT/cml-tts-filtered
Parler-TTS Mini 多语言版
Parler-TTS Mini 多语言版 v1 是 Parler-TTS Mini 的多语言扩展版本。
这是一个经过微调的版本,训练数据包括 CML-TTS 的 清洗版本 和 多语言LibriSpeech 的非英语部分。总计约9,200小时的非英语数据。为了保留英语能力,我们还添加了 LibriTTS-R英语数据集,约580小时的高质量英语数据。
Parler-TTS Mini 多语言版 支持8种欧洲语言:英语、法语、西班牙语、葡萄牙语、波兰语、德语、意大利语和荷兰语。
得益于其 改进的提示词标记器,它可以轻松扩展到其他语言。该标记器具有更大的词汇量并支持字节回退,简化了多语言训练。
🚨 这项工作是 HuggingFace音频团队 与 Quantum Squadra 团队 合作的成果。AI4Bharat 团队 也提供了改进标记化的建议和帮助。🚨
📖 快速索引
🛠️ 使用说明
🚨与之前版本的Parler-TTS不同,这里我们使用两个标记器 - 一个用于提示词,一个用于描述。🚨
👨💻 安装
使用Parler-TTS就像说"bonjour"一样简单。只需安装一次库:
pip install git+https://github.com/huggingface/parler-tts.git
推理
Parler-TTS 经过训练,可以通过简单的文本提示控制生成语音特征,例如:
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-multilingual").to(device)
tokenizer = AutoTokenizer.from_pretrained("parler-tts/parler-tts-mini-multilingual")
description_tokenizer = AutoTokenizer.from_pretrained(model.config.text_encoder._name_or_path)
prompt = "嘿,你今天过得怎么样?"
description = "一位女性演讲者以略带表现力和生动的语速和音高发表演讲。录音质量非常高,演讲者的声音清晰且非常近距离。"
input_ids = description_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)
提示:
- 我们设置了 推理指南 来加速生成。考虑使用SDPA、torch.compile、批处理和流式处理!
- 包含"非常清晰的音频"可以生成最高质量的音频,"非常嘈杂的音频"则会生成高背景噪音
- 可以使用标点符号控制生成的韵律,例如使用逗号添加小的停顿
- 其余语音特征(性别、语速、音高和混响)可以直接通过提示词控制
动机
Parler-TTS是对Stability AI和爱丁堡大学的Dan Lyth和Simon King发表的论文《使用合成标注的自然语言指导高保真文本转语音》工作的复现。
与其他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许可下获得许可。