deepspeed

游雁

2024-05-17 f68ae892be3e59c20a033f18d5f61db7f633801f

 runtime/onnxruntime/src/ct-transformer.h

@@ -26,7 +26,7 @@
public:

   CTTransformer();
   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);
   ~CTTransformer();
   vector<int>  Infer(vector<int32_t> input_data);
   string AddPunc(const char* sz_input, std::string language="zh-cn");