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

Add CI to build HAPs for HarmonyOS (#1578)

Browse Source
This commit is contained in:
Fangjun Kuang
2024-11-29 21:13:01 +08:00
committed by GitHub
parent be159f943e
commit 299f2392e2
11 changed files with 376 additions and 24 deletions
Download Patch File Download Diff File Expand all files Collapse all files

28
scripts/apk/generate-vad-asr-apk-script.py
View File

@@ -2,6 +2,7 @@
import argparse
from dataclasses import dataclass
from pathlib import Path
import jinja2
@@ -34,6 +35,7 @@ class Model:
# e.g., zh, en, zh_en
lang: str
lang2: str
# e.g., whisper, paraformer, zipformer
short_name: str = ""
@@ -51,6 +53,7 @@ def get_models():
model_name="sherpa-onnx-whisper-tiny.en",
idx=2,
lang="en",
lang2="English",
short_name="whisper_tiny",
cmd="""
pushd $model_name
@@ -71,6 +74,7 @@ def get_models():
model_name="sherpa-onnx-paraformer-zh-2023-09-14",
idx=0,
lang="zh_en",
lang2="Chinese,English",
short_name="paraformer",
rule_fsts="itn_zh_number.fst",
cmd="""
@@ -92,6 +96,7 @@ def get_models():
model_name="sherpa-onnx-sense-voice-zh-en-ja-ko-yue-2024-07-17",
idx=15,
lang="zh_en_ko_ja_yue",
lang2="中英粤日韩",
short_name="sense_voice",
cmd="""
pushd $model_name
@@ -109,6 +114,7 @@ def get_models():
model_name="sherpa-onnx-paraformer-zh-small-2024-03-09",
idx=14,
lang="zh_en",
lang2="Chinese,English",
short_name="small_paraformer",
rule_fsts="itn_zh_number.fst",
cmd="""
@@ -132,6 +138,7 @@ def get_models():
model_name="icefall-asr-zipformer-wenetspeech-20230615",
idx=4,
lang="zh",
lang2="Chinese",
short_name="zipformer",
rule_fsts="itn_zh_number.fst",
cmd="""
@@ -159,6 +166,7 @@ def get_models():
model_name="sherpa-onnx-nemo-fast-conformer-ctc-be-de-en-es-fr-hr-it-pl-ru-uk-20k",
idx=7,
lang="be_de_en_es_fr_hr_it_pl_ru_uk",
lang2="be_de_en_es_fr_hr_it_pl_ru_uk",
short_name="fast_conformer_ctc_20k",
cmd="""
pushd $model_name
@@ -174,6 +182,7 @@ def get_models():
model_name="sherpa-onnx-nemo-fast-conformer-ctc-en-24500",
idx=8,
lang="en",
lang2="English",
short_name="fast_conformer_ctc_24500",
cmd="""
pushd $model_name
@@ -188,7 +197,8 @@ def get_models():
Model(
model_name="sherpa-onnx-nemo-fast-conformer-ctc-en-de-es-fr-14288",
idx=9,
lang="en_des_es_fr",
lang="en_de_es_fr",
lang2="English,German,Spanish,French",
short_name="fast_conformer_ctc_14288",
cmd="""
pushd $model_name
@@ -204,6 +214,7 @@ def get_models():
model_name="sherpa-onnx-nemo-fast-conformer-ctc-es-1424",
idx=10,
lang="es",
lang2="Spanish",
short_name="fast_conformer_ctc_1424",
cmd="""
pushd $model_name
@@ -219,6 +230,7 @@ def get_models():
model_name="sherpa-onnx-telespeech-ctc-int8-zh-2024-06-04",
idx=11,
lang="zh",
lang2="Chinese",
short_name="telespeech",
rule_fsts="itn_zh_number.fst",
cmd="""
@@ -239,6 +251,7 @@ def get_models():
model_name="sherpa-onnx-zipformer-thai-2024-06-20",
idx=12,
lang="th",
lang2="Thai",
short_name="zipformer",
cmd="""
pushd $model_name
@@ -260,6 +273,7 @@ def get_models():
model_name="sherpa-onnx-zipformer-korean-2024-06-24",
idx=13,
lang="ko",
lang2="Korean",
short_name="zipformer",
cmd="""
pushd $model_name
@@ -281,6 +295,7 @@ def get_models():
model_name="sherpa-onnx-zipformer-ja-reazonspeech-2024-08-01",
idx=16,
lang="ja",
lang2="Japanese",
short_name="zipformer_reazonspeech",
cmd="""
pushd $model_name
@@ -300,6 +315,7 @@ def get_models():
model_name="sherpa-onnx-zipformer-ru-2024-09-18",
idx=17,
lang="ru",
lang2="Russian",
short_name="zipformer",
cmd="""
pushd $model_name
@@ -320,6 +336,7 @@ def get_models():
model_name="sherpa-onnx-small-zipformer-ru-2024-09-18",
idx=18,
lang="ru",
lang2="Russian",
short_name="small_zipformer",
cmd="""
pushd $model_name
@@ -340,6 +357,7 @@ def get_models():
model_name="sherpa-onnx-nemo-ctc-giga-am-russian-2024-10-24",
idx=19,
lang="ru",
lang2="Russian",
short_name="nemo_ctc_giga_am",
cmd="""
pushd $model_name
@@ -358,6 +376,7 @@ def get_models():
model_name="sherpa-onnx-nemo-transducer-giga-am-russian-2024-10-24",
idx=20,
lang="ru",
lang2="Russian",
short_name="nemo_transducer_giga_am",
cmd="""
pushd $model_name
@@ -376,6 +395,7 @@ def get_models():
model_name="sherpa-onnx-moonshine-tiny-en-int8",
idx=21,
lang="en",
lang2="English",
short_name="moonshine_tiny_int8",
cmd="""
pushd $model_name
@@ -391,6 +411,7 @@ def get_models():
model_name="sherpa-onnx-moonshine-base-en-int8",
idx=22,
lang="en",
lang2="English",
short_name="moonshine_base_int8",
cmd="""
pushd $model_name
@@ -436,9 +457,14 @@ def main():
filename_list = [
"./build-apk-vad-asr.sh",
"./build-hap-vad-asr.sh",
]
for filename in filename_list:
environment = jinja2.Environment()
if not Path(f"{filename}.in").is_file():
print(f"skip {filename}")
continue
with open(f"{filename}.in") as f:
s = f.read()
template = environment.from_string(s)