Add APK for small paraformer (#1133)
This commit is contained in:
@@ -89,6 +89,29 @@ def get_models():
|
|||||||
popd
|
popd
|
||||||
""",
|
""",
|
||||||
),
|
),
|
||||||
|
Model(
|
||||||
|
model_name="sherpa-onnx-paraformer-zh-small-2024-03-09",
|
||||||
|
idx=14,
|
||||||
|
lang="zh",
|
||||||
|
short_name="small_paraformer",
|
||||||
|
rule_fsts="itn_zh_number.fst",
|
||||||
|
cmd="""
|
||||||
|
if [ ! -f itn_zh_number.fst ]; then
|
||||||
|
curl -SL -O https://github.com/k2-fsa/sherpa-onnx/releases/download/asr-models/itn_zh_number.fst
|
||||||
|
fi
|
||||||
|
pushd $model_name
|
||||||
|
|
||||||
|
rm -fv README.md
|
||||||
|
rm -fv *.py
|
||||||
|
rm -fv *.yaml
|
||||||
|
rm -fv *.mvn
|
||||||
|
rm -rfv test_wavs
|
||||||
|
|
||||||
|
ls -lh
|
||||||
|
|
||||||
|
popd
|
||||||
|
""",
|
||||||
|
),
|
||||||
Model(
|
Model(
|
||||||
model_name="icefall-asr-zipformer-wenetspeech-20230615",
|
model_name="icefall-asr-zipformer-wenetspeech-20230615",
|
||||||
idx=4,
|
idx=4,
|
||||||
|
|||||||
@@ -310,6 +310,17 @@ fun getOfflineModelConfig(type: Int): OfflineModelConfig? {
|
|||||||
modelType = "transducer",
|
modelType = "transducer",
|
||||||
)
|
)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
14 -> {
|
||||||
|
val modelDir = "sherpa-onnx-paraformer-zh-small-2024-03-09"
|
||||||
|
return OfflineModelConfig(
|
||||||
|
paraformer = OfflineParaformerModelConfig(
|
||||||
|
model = "$modelDir/model.int8.onnx",
|
||||||
|
),
|
||||||
|
tokens = "$modelDir/tokens.txt",
|
||||||
|
modelType = "paraformer",
|
||||||
|
)
|
||||||
|
}
|
||||||
}
|
}
|
||||||
return null
|
return null
|
||||||
}
|
}
|
||||||
|
|||||||
Reference in New Issue
Block a user