From 0535db1c65180cfb4da046c5d865c764e6445746 Mon Sep 17 00:00:00 2001
From: lyblsgo <lyblsgo@163.com>
Date: 星期一, 24 四月 2023 10:57:53 +0800
Subject: [PATCH] rename CT-transformer
---
funasr/runtime/onnxruntime/src/paraformer_onnx.cpp | 5 ++---
1 files changed, 2 insertions(+), 3 deletions(-)
diff --git a/funasr/runtime/onnxruntime/src/paraformer_onnx.cpp b/funasr/runtime/onnxruntime/src/paraformer_onnx.cpp
index 69d1554..289eab1 100644
--- a/funasr/runtime/onnxruntime/src/paraformer_onnx.cpp
+++ b/funasr/runtime/onnxruntime/src/paraformer_onnx.cpp
@@ -3,7 +3,7 @@
using namespace std;
using namespace paraformer;
-ModelImp::ModelImp(const char* path,int nNumThread, bool quantize, bool use_vad)
+ModelImp::ModelImp(const char* path,int nNumThread, bool quantize, bool use_vad, bool use_punc)
:env_(ORT_LOGGING_LEVEL_ERROR, "paraformer"),sessionOptions{}{
string model_path;
string cmvn_path;
@@ -18,7 +18,7 @@
}
// PUNC model
- if(true){
+ if(use_punc){
puncHandle = make_unique<CTTransformer>(path, nNumThread);
}
@@ -55,7 +55,6 @@
m_session = std::make_unique<Ort::Session>(env_, model_path.c_str(), sessionOptions);
#endif
- vector<string> m_strInputNames, m_strOutputNames;
string strName;
getInputName(m_session.get(), strName);
m_strInputNames.push_back(strName.c_str());
--
Gitblit v1.9.1