From de3f9462e020ad2f4c7302ec07fa808d177cec95 Mon Sep 17 00:00:00 2001
From: 夜雨飘零 <yeyupiaoling@foxmail.com>
Date: 星期三, 18 十月 2023 16:20:35 +0800
Subject: [PATCH] change send data size (#1014)
---
funasr/runtime/onnxruntime/include/tpass-stream.h | 9 ++++++++-
1 files changed, 8 insertions(+), 1 deletions(-)
diff --git a/funasr/runtime/onnxruntime/include/tpass-stream.h b/funasr/runtime/onnxruntime/include/tpass-stream.h
index f9a5385..0276631 100644
--- a/funasr/runtime/onnxruntime/include/tpass-stream.h
+++ b/funasr/runtime/onnxruntime/include/tpass-stream.h
@@ -7,6 +7,9 @@
#include "model.h"
#include "punc-model.h"
#include "vad-model.h"
+#if !defined(__APPLE__)
+#include "itn-model.h"
+#endif
namespace funasr {
class TpassStream {
@@ -14,16 +17,20 @@
TpassStream(std::map<std::string, std::string>& model_path, int thread_num);
~TpassStream(){};
- // std::unique_ptr<VadModel> vad_handle = nullptr;
std::unique_ptr<VadModel> vad_handle = nullptr;
std::unique_ptr<Model> asr_handle = nullptr;
std::unique_ptr<PuncModel> punc_online_handle = nullptr;
+#if !defined(__APPLE__)
+ std::unique_ptr<ITNModel> itn_handle = nullptr;
+#endif
bool UseVad(){return use_vad;};
bool UsePunc(){return use_punc;};
+ bool UseITN(){return use_itn;};
private:
bool use_vad=false;
bool use_punc=false;
+ bool use_itn=false;
};
TpassStream *CreateTpassStream(std::map<std::string, std::string>& model_path, int thread_num=1);
--
Gitblit v1.9.1