From 687277ae9d864f33819bc6fe33584271d48776d3 Mon Sep 17 00:00:00 2001
From: 雾聪 <wucong.lyb@alibaba-inc.com>
Date: 星期一, 01 四月 2024 11:27:26 +0800
Subject: [PATCH] update funasrruntime
---
runtime/onnxruntime/src/funasrruntime.cpp | 4 ++--
1 files changed, 2 insertions(+), 2 deletions(-)
diff --git a/runtime/onnxruntime/src/funasrruntime.cpp b/runtime/onnxruntime/src/funasrruntime.cpp
index e9a3f59..e1b1477 100644
--- a/runtime/onnxruntime/src/funasrruntime.cpp
+++ b/runtime/onnxruntime/src/funasrruntime.cpp
@@ -252,7 +252,7 @@
if (wfst_decoder){
wfst_decoder->StartUtterance();
}
- vector<string> msgs = (offline_stream->asr_handle)->Forward(buff, len, true, hw_emb, dec_handle);
+ vector<string> msgs = (offline_stream->asr_handle)->Forward(buff, len, true, hw_emb, dec_handle, batch_in);
for(int idx=0; idx<batch_in; idx++){
string msg = msgs[idx];
std::vector<std::string> msg_vec = funasr::split(msg, '|');
@@ -359,7 +359,7 @@
if (wfst_decoder){
wfst_decoder->StartUtterance();
}
- vector<string> msgs = (offline_stream->asr_handle)->Forward(buff, len, true, hw_emb, dec_handle);
+ vector<string> msgs = (offline_stream->asr_handle)->Forward(buff, len, true, hw_emb, dec_handle, batch_in);
for(int idx=0; idx<batch_in; idx++){
string msg = msgs[idx];
std::vector<std::string> msg_vec = funasr::split(msg, '|');
--
Gitblit v1.9.1