From 1819303f5e8cfc03f4c0ec2495571a54a186d34b Mon Sep 17 00:00:00 2001
From: 雾聪 <wucong.lyb@alibaba-inc.com>
Date: 星期二, 29 十月 2024 11:40:18 +0800
Subject: [PATCH] support SenseVoiceSmall in 2pass mode
---
runtime/onnxruntime/src/paraformer.cpp | 27 +++------------------------
1 files changed, 3 insertions(+), 24 deletions(-)
diff --git a/runtime/onnxruntime/src/paraformer.cpp b/runtime/onnxruntime/src/paraformer.cpp
index 24f5152..7e1fe96 100644
--- a/runtime/onnxruntime/src/paraformer.cpp
+++ b/runtime/onnxruntime/src/paraformer.cpp
@@ -131,9 +131,10 @@
}
// 2pass
-void Paraformer::InitAsr(const std::string &am_model, const std::string &en_model, const std::string &de_model, const std::string &am_cmvn, const std::string &am_config, const std::string &token_file, int thread_num){
+void Paraformer::InitAsr(const std::string &am_model, const std::string &en_model, const std::string &de_model,
+ const std::string &am_cmvn, const std::string &am_config, const std::string &token_file, const std::string &online_token_file, int thread_num){
// online
- InitAsr(en_model, de_model, am_cmvn, am_config, token_file, thread_num);
+ InitAsr(en_model, de_model, am_cmvn, am_config, online_token_file, thread_num);
// offline
try {
@@ -144,28 +145,6 @@
exit(-1);
}
- // string strName;
- // GetInputName(m_session_.get(), strName);
- // m_strInputNames.push_back(strName.c_str());
- // GetInputName(m_session_.get(), strName,1);
- // m_strInputNames.push_back(strName);
-
- // if (use_hotword) {
- // GetInputName(m_session_.get(), strName, 2);
- // m_strInputNames.push_back(strName);
- // }
-
- // // support time stamp
- // size_t numOutputNodes = m_session_->GetOutputCount();
- // for(int index=0; index<numOutputNodes; index++){
- // GetOutputName(m_session_.get(), strName, index);
- // m_strOutputNames.push_back(strName);
- // }
-
- // for (auto& item : m_strInputNames)
- // m_szInputNames.push_back(item.c_str());
- // for (auto& item : m_strOutputNames)
- // m_szOutputNames.push_back(item.c_str());
GetInputNames(m_session_.get(), m_strInputNames, m_szInputNames);
GetOutputNames(m_session_.get(), m_strOutputNames, m_szOutputNames);
}
--
Gitblit v1.9.1