EngineX-Iluvatar/enginex_bi_series-sherpa-onnx
Archived
16
0
Fork 0
Code Issues Pull Requests Actions Projects Releases Wiki Activity

Add HarmonyOS demo for Kokoro TTS 1.0 (#1813)

Browse Source
This commit is contained in:
Fangjun Kuang
2025-02-07 18:43:26 +08:00
committed by GitHub
parent 239b43c278
commit 5ca246559b
2 changed files with 40 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
harmony-os/SherpaOnnxHar/sherpa_onnx/src/main/ets/components/NonStreamingTts.ets
View File

@@ -34,6 +34,8 @@ export class OfflineTtsKokoroModelConfig {
public tokens: string = ''; public tokens: string = '';
public dataDir: string = ''; public dataDir: string = '';
public lengthScale: number = 1.0; public lengthScale: number = 1.0;
public dictDir: string = '';
public lexicon: string = '';
} }
export class OfflineTtsModelConfig { export class OfflineTtsModelConfig {

52
harmony-os/SherpaOnnxTts/entry/src/main/ets/workers/NonStreamingTtsWorker.ets
View File

@@ -179,6 +179,16 @@ function initTts(context: Context): OfflineTts {
// voices = 'voices.bin' // voices = 'voices.bin'
// dataDir = 'espeak-ng-data'; // dataDir = 'espeak-ng-data';
// Example 11
// https://k2-fsa.github.io/sherpa/onnx/tts/pretrained_models/kokoro.html
// modelDir = 'kokoro-multi-lang-v1_0';
// modelName = 'model.onnx';
// voices = 'voices.bin'
// dataDir = 'espeak-ng-data';
// dictDir = 'dict';
// lexicon = 'lexicon-us-en.txt,lexicon-zh.txt';
// ruleFsts = `date-zh.fst,phone-zh.fst,number-zh.fst`;
// ============================================================ // ============================================================
// Please don't change the remaining part of this function // Please don't change the remaining part of this function
// ============================================================ // ============================================================
@@ -225,7 +235,14 @@ function initTts(context: Context): OfflineTts {
ruleFars = tmp.join(','); ruleFars = tmp.join(',');
} }
if (lexicon != '') { if (lexicon.includes(",")) {
let v = lexicon.split(',')
let tmp: string[] = [];
for (const f of v) {
tmp.push(modelDir + '/' + f);
}
lexicon = tmp.join(',');
} else if (lexicon != '') {
lexicon = modelDir + '/' + lexicon; lexicon = modelDir + '/' + lexicon;
} }
@@ -250,26 +267,33 @@ function initTts(context: Context): OfflineTts {
config.model.vits.model = modelName; config.model.vits.model = modelName;
} }
config.model.vits.lexicon = lexicon; if (voices == '') {
config.model.vits.tokens = tokens; config.model.vits.lexicon = lexicon;
config.model.vits.dataDir = dataDir; config.model.vits.tokens = tokens;
config.model.vits.dictDir = dictDir; config.model.vits.dataDir = dataDir;
config.model.vits.dictDir = dictDir;
config.model.matcha.acousticModel = acousticModelName; config.model.matcha.acousticModel = acousticModelName;
config.model.matcha.vocoder = vocoder; config.model.matcha.vocoder = vocoder;
config.model.matcha.lexicon = lexicon; config.model.matcha.lexicon = lexicon;
config.model.matcha.tokens = tokens; config.model.matcha.tokens = tokens;
config.model.matcha.dataDir = dataDir; config.model.matcha.dataDir = dataDir;
config.model.matcha.dictDir = dictDir; config.model.matcha.dictDir = dictDir;
}
if (voices != '') { if (voices != '') {
config.model.kokoro.model = modelName; config.model.kokoro.model = modelName;
} else { } else {
config.model.kokoro.model = ''; config.model.kokoro.model = '';
} }
config.model.kokoro.voices = voices;
config.model.kokoro.tokens = tokens; if (voices != '') {
config.model.kokoro.dataDir = dataDir; config.model.kokoro.voices = voices;
config.model.kokoro.tokens = tokens;
config.model.kokoro.dataDir = dataDir;
config.model.kokoro.dictDir = dictDir;
config.model.kokoro.lexicon = lexicon;
}
config.model.numThreads = 2; config.model.numThreads = 2;
config.model.debug = true; config.model.debug = true;