Support the latest offline zipformer model (#180)

2023-06-19 14:09:24 +08:00
parent 8562711252
commit d82251cba6
1 changed files with 2 additions and 1 deletions
--- a/sherpa-onnx/csrc/offline-recognizer-impl.cc
+++ b/sherpa-onnx/csrc/offline-recognizer-impl.cc
@@ -67,7 +67,8 @@ std::unique_ptr<OfflineRecognizerImpl> OfflineRecognizerImpl::Create(
  }
  std::string model_type(model_type_ptr.get());

-  if (model_type == "conformer" || model_type == "zipformer") {
+  if (model_type == "conformer" || model_type == "zipformer" ||
+      model_type == "zipformer2") {
    return std::make_unique<OfflineRecognizerTransducerImpl>(config);
  }