From b9bcf1f093c3053fdc4e2cf4a1d38e27bbf429fb Mon Sep 17 00:00:00 2001
From: 游雁 <zhifu.gzf@alibaba-inc.com>
Date: 星期四, 19 十月 2023 14:03:48 +0800
Subject: [PATCH] docs
---
funasr/runtime/onnxruntime/src/offline-stream.cpp | 18 +++++++++++++++++-
1 files changed, 17 insertions(+), 1 deletions(-)
diff --git a/funasr/runtime/onnxruntime/src/offline-stream.cpp b/funasr/runtime/onnxruntime/src/offline-stream.cpp
index c042562..caea72e 100644
--- a/funasr/runtime/onnxruntime/src/offline-stream.cpp
+++ b/funasr/runtime/onnxruntime/src/offline-stream.cpp
@@ -1,5 +1,4 @@
#include "precomp.h"
-#include <unistd.h>
namespace funasr {
OfflineStream::OfflineStream(std::map<std::string, std::string>& model_path, int thread_num)
@@ -84,6 +83,23 @@
use_punc = true;
}
}
+#if !defined(__APPLE__)
+ // Optional: ITN, here we just support language_type=MandarinEnglish
+ if(model_path.find(ITN_DIR) != model_path.end() && model_path.at(ITN_DIR) != ""){
+ string itn_tagger_path = PathAppend(model_path.at(ITN_DIR), ITN_TAGGER_NAME);
+ string itn_verbalizer_path = PathAppend(model_path.at(ITN_DIR), ITN_VERBALIZER_NAME);
+
+ if (access(itn_tagger_path.c_str(), F_OK) != 0 ||
+ access(itn_verbalizer_path.c_str(), F_OK) != 0 )
+ {
+ LOG(INFO) << "ITN model file is not exist, skip load ITN model.";
+ }else{
+ itn_handle = make_unique<ITNProcessor>();
+ itn_handle->InitITN(itn_tagger_path, itn_verbalizer_path, thread_num);
+ use_itn = true;
+ }
+ }
+#endif
}
OfflineStream *CreateOfflineStream(std::map<std::string, std::string>& model_path, int thread_num)
--
Gitblit v1.9.1