FUNASR训练
blame | 历史 | 补丁 | 提交 | 提交对比 | ignore whitespace
雾聪
2023-09-28 a49f2c6411637d696e787605ec611f05667e8935 
 funasr/runtime/onnxruntime/include/tpass-stream.h


@@ -7,7 +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 {


@@ -18,7 +20,9 @@


    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;};