🚀 Parler-TTS Mini v0.1
Parler-TTS Mini v0.1 是一款輕量級的文本轉語音(TTS)模型,基於 10500 小時的音頻數據進行訓練。它能夠生成高質量、自然流暢的語音,並且可以通過簡單的文本提示(例如性別、背景噪音、語速、音調和混響)來控制語音特徵。該模型是 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 = "Hey, how are you doing today?"
description = "A female speaker with a slightly low-pitched voice delivers her words quite expressively, in a very confined sounding environment with clear audio quality. She speaks very fast."
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)
使用提示
💡 使用建議
- 包含 “very clear audio” 一詞可以生成最高質量的音頻,包含 “very noisy audio” 則可以生成具有高背景噪音的音頻。
- 可以使用標點符號來控制生成語音的韻律,例如使用逗號在語音中添加小停頓。
- 其餘語音特徵(性別、語速、音調和混響)可以直接通過提示進行控制。
✨ 主要特性
- 輕量級:Parler-TTS Mini v0.1 是一款輕量級的 TTS 模型,便於使用和部署。
- 高質量語音生成:基於大量音頻數據訓練,能夠生成高質量、自然流暢的語音。
- 特徵可控制:可以通過簡單的文本提示控制語音的多種特徵,如性別、背景噪音、語速、音調和混響等。
📚 詳細文檔
模型動機
Parler-TTS 是對 Dan Lyth(來自 Stability AI)和 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 許可證。
📦 訓練數據集
屬性 |
詳情 |
訓練數據 |
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 |