From 9b4e0969f29dccdfa78a55219f8107e549c96e45 Mon Sep 17 00:00:00 2001
From: lzr265946 <lzr265946@alibaba-inc.com>
Date: 星期四, 07 九月 2023 17:19:02 +0800
Subject: [PATCH] fix transformerLM inference recipe
---
funasr/runtime/onnxruntime/include/offline-stream.h | 8 +++++---
1 files changed, 5 insertions(+), 3 deletions(-)
diff --git a/funasr/runtime/onnxruntime/include/offline-stream.h b/funasr/runtime/onnxruntime/include/offline-stream.h
index caa4ea6..bc19bf2 100644
--- a/funasr/runtime/onnxruntime/include/offline-stream.h
+++ b/funasr/runtime/onnxruntime/include/offline-stream.h
@@ -8,14 +8,15 @@
#include "punc-model.h"
#include "vad-model.h"
+namespace funasr {
class OfflineStream {
public:
OfflineStream(std::map<std::string, std::string>& model_path, int thread_num);
~OfflineStream(){};
- std::unique_ptr<VadModel> vad_handle;
- std::unique_ptr<Model> asr_handle;
- std::unique_ptr<PuncModel> punc_handle;
+ std::unique_ptr<VadModel> vad_handle= nullptr;
+ std::unique_ptr<Model> asr_handle= nullptr;
+ std::unique_ptr<PuncModel> punc_handle= nullptr;
bool UseVad(){return use_vad;};
bool UsePunc(){return use_punc;};
@@ -25,4 +26,5 @@
};
OfflineStream *CreateOfflineStream(std::map<std::string, std::string>& model_path, int thread_num=1);
+} // namespace funasr
#endif
--
Gitblit v1.9.1