add zipformer model_type (#97)

Co-authored-by: lichao <www.563042811@qq.com>
This commit is contained in:
Jason's Lab
2023-03-29 20:36:39 +08:00
committed by GitHub
parent 423d89e9a5
commit 5e5620ea23

View File

@@ -42,7 +42,7 @@ std::unique_ptr<OfflineRecognizerImpl> OfflineRecognizerImpl::Create(
std::string model_type;
SHERPA_ONNX_READ_META_DATA_STR(model_type, "model_type");
if (model_type == "conformer") {
if (model_type == "conformer" || model_type == "zipformer") {
return std::make_unique<OfflineRecognizerTransducerImpl>(config);
}