VietTTS là một bộ công cụ mã nguồn mở cung cấp mô hình TTS tiếng Việt mạnh mẽ, cho phép tổng hợp giọng nói tự nhiên và tạo giọng nói mới. VietTTS hỗ trợ nghiên cứu và ứng dụng trong công nghệ giọng nói tiếng Việt.
- TTS: Tổng hợp giọng nói từ văn bản với bất kỳ giọng nào qua audio mẫu
- OpenAI-API-compatible: Tương thích với API Text to Speech OpenAI
VietTTS có thể được cài đặt qua trình cài đặt Python (chỉ hỗ trợ Linux, Windows và macOS sẽ có trong tương lai) hoặc Docker.
git clone https://github.com/dangvansam/viet-tts.git
cd viet-tts
# (Tùy chọn) Tạo môi trường Python với conda hoặc dùng virtualenv
conda create --name viettts python=3.10
conda activate viettts
# Cài đặt
pip install -e . && pip cache purge
-
Cài đặt Docker, NVIDIA Driver, NVIDIA Container Toolkit, và CUDA.
-
Chạy các lệnh sau:
git clone https://github.com/dangvansam/viet-tts.git
cd viet-tts
# Xây dựng hình ảnh docker
docker compose build
# Chạy bằng docker-compose - tạo server tại: http://localhost:8298
docker compose up -d
# Chạy bằng docker run - tạo server tại: http://localhost:8298
docker run -itd --gpu=alls -p 8298:8298 -v ./pretrained-models:/app/pretrained-models -n viet-tts-service viet-tts:latest viettts server --host 0.0.0.0 --port 8298
Bạn có thể sử dụng các giọng nói có sẵn dưới đây để tổng hợp giọng nói.
Mở rộng
ID | Giọng | Giới tính | Phát âm thanh |
---|---|---|---|
1 | nsnd-le-chuc | 👨 | |
2 | speechify_10 | 👩 | |
3 | atuan | 👨 | |
4 | speechify_11 | 👩 | |
5 | cdteam | 👨 | |
6 | speechify_12 | 👩 | |
7 | cross_lingual_prompt | 👩 | |
8 | speechify_2 | 👩 | |
9 | diep-chi | 👨 | |
10 | speechify_3 | 👩 | |
11 | doremon | 👨 | |
12 | speechify_4 | 👩 | |
13 | jack-sparrow | 👨 | |
14 | speechify_5 | 👩 | |
15 | nguyen-ngoc-ngan | 👩 | |
16 | speechify_6 | 👩 | |
17 | nu-nhe-nhang | 👩 | |
18 | speechify_7 | 👩 | |
19 | quynh | 👩 | |
20 | speechify_8 | 👩 | |
21 | speechify_9 | 👩 | |
22 | son-tung-mtp | 👨 | |
23 | zero_shot_prompt | 👩 | |
24 | speechify_1 | 👩 |
Giao diện dòng lệnh VietTTS cho phép bạn tạo giọng nói từ terminal. Cách sử dụng:
# Hướng dẫn sử dụng
viettts --help
# Khởi động API Server
viettts server --host 0.0.0.0 --port 8298
# Xem tất cả các giọng nói có sẵn
viettts show-voices
# Tổng hợp giọng nói từ văn bản với giọng có sẵn
viettts synthesis --text "Xin chào" --voice 0 --output test.wav
# Sao chép giọng từ audio file bất kì
viettts synthesis --text "Xin chào" --voice Download/voice.wav --output cloned.wav
Thiết lập biến môi trường cho OpenAI Client:
# Thiết lập base_url và API key như biến môi trường
export OPENAI_BASE_URL=http://localhost:8298
export OPENAI_API_KEY=viet-tts # không dùng trong phiên bản hiện tại
Để tạo giọng nói từ văn bản đầu vào:
from pathlib import Path
from openai import OpenAI
client = OpenAI()
output_file_path = Path(__file__).parent / "speech.wav"
with client.audio.speech.with_streaming_response.create(
model='tts-1',
voice='cdteam',
input='Xin chào Việt Nam.',
speed=1.0,
response_format='wav'
) as response:
response.stream_to_file('a.wav')
# Lấy danh sách giọng có sẵn
curl --location http://0.0.0.0:8298/v1/voices
# OpenAI API format
curl http://localhost:8298/v1/audio/speech \
-H "Authorization: Bearer viet-tts" \
-H "Content-Type: application/json" \
-d '{
"model": "tts-1",
"input": "Xin chào Việt Nam.",
"voice": "son-tung-mtp"
}' \
--output speech.wav
# API với giọng từ file local
curl --location http://0.0.0.0:8298/v1/tts \
--form 'text="xin chào"' \
--form 'audio_file=@"/home/viettts/Downloads/voice.mp4"' \
--output speech.wav
import fs from "fs";
import path from "path";
import OpenAI from "openai";
const openai = new OpenAI();
const speechFile = path.resolve("./speech.wav");
async function main() {
const mp3 = await openai.audio.speech.create({
model: "tts-1",
voice: "1",
input: "Xin chào Việt Nam.",
});
console.log(speechFile);
const buffer = Buffer.from(await mp3.arrayBuffer());
await fs.promises.writeFile(speechFile, buffer);
}
main();
- 💡 Sử dụng mã từ Cosyvoice
- 🎙️ Mô hình VAD từ silero-vad
- 📝 Chuẩn hóa văn bản với Vinorm
Mã nguồn của VietTTS được cấp phép theo Apache 2.0 License. Mô hình và mẫu âm thanh huấn luyện được cấp phép theo CC BY-NC License, dựa trên tập dữ liệu từ internet. Xin lỗi nếu điều này gây bất tiện.
Nội dung trên chỉ phục vụ mục đích học thuật và nhằm trình bày khả năng kỹ thuật. Một số ví dụ lấy từ internet. Nếu nội dung vi phạm quyền của bạn, vui lòng liên hệ để được gỡ bỏ.
- Facebook: https://fb.com/sam.rngd
- GitHub: https://github.com/dangvansam
- Email: [email protected]