40 lines
1.2 KiB
JavaScript
40 lines
1.2 KiB
JavaScript
// Copyright (c) 2023 Xiaomi Corporation (authors: Fangjun Kuang)
|
|
|
|
const sherpa_onnx = require('sherpa-onnx');
|
|
|
|
function createOfflineTts() {
|
|
let offlineTtsVitsModelConfig = {
|
|
model: './vits-icefall-zh-aishell3/model.onnx',
|
|
lexicon: './vits-icefall-zh-aishell3/lexicon.txt',
|
|
tokens: './vits-icefall-zh-aishell3/tokens.txt',
|
|
noiseScale: 0.667,
|
|
noiseScaleW: 0.8,
|
|
lengthScale: 1.0,
|
|
};
|
|
let offlineTtsModelConfig = {
|
|
offlineTtsVitsModelConfig: offlineTtsVitsModelConfig,
|
|
numThreads: 1,
|
|
debug: 1,
|
|
provider: 'cpu',
|
|
};
|
|
|
|
let offlineTtsConfig = {
|
|
offlineTtsModelConfig: offlineTtsModelConfig,
|
|
ruleFsts:
|
|
'./vits-icefall-zh-aishell3/phone.fst,./vits-icefall-zh-aishell3/date.fst,./vits-icefall-zh-aishell3/number.fst,./vits-icefall-zh-aishell3/new_heteronym.fst',
|
|
ruleFars: './vits-icefall-zh-aishell3/rule.far',
|
|
maxNumSentences: 1,
|
|
};
|
|
|
|
return sherpa_onnx.createOfflineTts(offlineTtsConfig);
|
|
}
|
|
|
|
const tts = createOfflineTts();
|
|
const speakerId = 66;
|
|
const speed = 1.0;
|
|
const audio = tts.generate(
|
|
{text: '3年前中国总人口是1411778724人', sid: speakerId, speed: speed});
|
|
tts.save('./test-vits-zh.wav', audio);
|
|
console.log('Saved to test-vits-zh.wav successfully.');
|
|
tts.free();
|