Quellcode durchsuchen

modify paraformer-online funcs

雾聪 vor 2 Jahren
Ursprung
Commit
dfe78d1877

+ 1 - 1
runtime/onnxruntime/src/paraformer-online.cpp

@@ -469,7 +469,7 @@ string ParaformerOnline::ForwardChunk(std::vector<std::vector<float>> &chunk_fea
     return result;
 }
 
-string ParaformerOnline::Forward(float* din, int len, bool input_finished, const std::vector<std::vector<float>> &hw_emb)
+string ParaformerOnline::Forward(float* din, int len, bool input_finished, const std::vector<std::vector<float>> &hw_emb, void* wfst_decoder)
 {
     std::vector<std::vector<float>> wav_feats;
     std::vector<float> waves(din, din+len);

+ 1 - 1
runtime/onnxruntime/src/paraformer-online.h

@@ -109,7 +109,7 @@ namespace funasr {
         void AddOverlapChunk(std::vector<std::vector<float>> &wav_feats, bool input_finished);
         
         string ForwardChunk(std::vector<std::vector<float>> &wav_feats, bool input_finished);
-        string Forward(float* din, int len, bool input_finished, const std::vector<std::vector<float>> &hw_emb={{0.0}});
+        string Forward(float* din, int len, bool input_finished, const std::vector<std::vector<float>> &hw_emb={{0.0}}, void* wfst_decoder=nullptr);
         string Rescoring();
         // 2pass
         std::string online_res;