FUNASR训练
blame | 历史 | 补丁 | 提交 | 提交对比 | ignore whitespace
游雁
2024-05-15 67733a2a92f54d9777a0b18ac4f1488b3bc41efc 
 runtime/onnxruntime/src/ct-transformer-online.h


@@ -26,7 +26,7 @@


public:




   CTTransformerOnline();


   void InitPunc(const std::string &punc_model, const std::string &punc_config, int thread_num);


   void InitPunc(const std::string &punc_model, const std::string &punc_config, const std::string &token_file, int thread_num);


   ~CTTransformerOnline();


   vector<int>  Infer(vector<int32_t> input_data, int nCacheSize);


   string AddPunc(const char* sz_input, vector<string> &arr_cache, std::string language="zh-cn");