From ab653d3871f72f7f6cd1ac3126b3df722f4c7943 Mon Sep 17 00:00:00 2001
From: aky15 <ankeyu.aky@11.17.44.249>
Date: 星期五, 20 十月 2023 15:33:09 +0800
Subject: [PATCH] add rwkv encoder

---
 funasr/runtime/onnxruntime/src/paraformer.h |    4 +++-
 1 files changed, 3 insertions(+), 1 deletions(-)

diff --git a/funasr/runtime/onnxruntime/src/paraformer.h b/funasr/runtime/onnxruntime/src/paraformer.h
index bac8fad..455078e 100644
--- a/funasr/runtime/onnxruntime/src/paraformer.h
+++ b/funasr/runtime/onnxruntime/src/paraformer.h
@@ -33,7 +33,6 @@
         vector<const char*> hw_m_szInputNames;
         vector<const char*> hw_m_szOutputNames;
         bool use_hotword;
-        std::string language="zh-cn";
 
     public:
         Paraformer();
@@ -55,6 +54,7 @@
         string PostProcess(std::vector<string> &raw_char, std::vector<std::vector<float>> &timestamp_list);
 
         string Rescoring();
+        string GetLang(){return language;};
 
         knf::FbankOptions fbank_opts_;
         vector<float> means_list_;
@@ -71,6 +71,8 @@
         vector<const char*> m_szInputNames;
         vector<const char*> m_szOutputNames;
 
+        std::string language="zh-cn";
+
         // paraformer-online
         std::shared_ptr<Ort::Session> encoder_session_ = nullptr;
         std::shared_ptr<Ort::Session> decoder_session_ = nullptr;

--
Gitblit v1.9.1