From a05e753d11d9c36983ec4e58c421dbcf86d1dcd4 Mon Sep 17 00:00:00 2001 From: Xian Shi <40013335+R1ckShi@users.noreply.github.com> Date: 星期二, 17 十月 2023 16:47:27 +0800 Subject: [PATCH] Merge branch 'main' into dev_onnx --- funasr/runtime/onnxruntime/include/com-define.h | 13 +++++++++++++ 1 files changed, 13 insertions(+), 0 deletions(-) diff --git a/funasr/runtime/onnxruntime/include/com-define.h b/funasr/runtime/onnxruntime/include/com-define.h index a1a5e0b..0ecb9c2 100644 --- a/funasr/runtime/onnxruntime/include/com-define.h +++ b/funasr/runtime/onnxruntime/include/com-define.h @@ -27,6 +27,7 @@ #define TXT_PATH "txt-path" #define THREAD_NUM "thread-num" #define PORT_ID "port-id" +#define HOTWORD_SEP " " // #define VAD_MODEL_PATH "vad-model" // #define VAD_CMVN_PATH "vad-cmvn" @@ -38,12 +39,20 @@ // #define PUNC_CONFIG_PATH "punc-config" #define MODEL_NAME "model.onnx" +// hotword embedding compile model +#define MODEL_EB_NAME "model_eb.onnx" #define QUANT_MODEL_NAME "model_quant.onnx" #define VAD_CMVN_NAME "vad.mvn" #define VAD_CONFIG_NAME "vad.yaml" #define AM_CMVN_NAME "am.mvn" #define AM_CONFIG_NAME "config.yaml" #define PUNC_CONFIG_NAME "punc.yaml" +#define MODEL_SEG_DICT "seg_dict" +#define HOTWORD "hotword" + +#define ITN_DIR "itn-dir" +#define ITN_TAGGER_NAME "zh_itn_tagger.fst" +#define ITN_VERBALIZER_NAME "zh_itn_verbalizer.fst" #define ENCODER_NAME "model.onnx" #define QUANT_ENCODER_NAME "model_quant.onnx" @@ -98,4 +107,8 @@ #define DUN_INDEX 5 #define CACHE_POP_TRIGGER_LIMIT 200 +#define JIEBA_DICT "jieba.c.dict" +#define JIEBA_USERDICT "jieba_usr_dict" +#define JIEBA_HMM_MODEL "jieba.hmm" + } // namespace funasr -- Gitblit v1.9.1