Yabin Li
2023-08-08 b454a1054fadbff0ee963944ff42f66b98317582
funasr/runtime/onnxruntime/include/com-define.h
@@ -13,11 +13,14 @@
// parser option
#define MODEL_DIR "model-dir"
#define OFFLINE_MODEL_DIR "model-dir"
#define ONLINE_MODEL_DIR "online-model-dir"
#define VAD_DIR "vad-dir"
#define PUNC_DIR "punc-dir"
#define QUANTIZE "quantize"
#define VAD_QUANT "vad-quant"
#define PUNC_QUANT "punc-quant"
#define ASR_MODE "mode"
#define WAV_PATH "wav-path"
#define WAV_SCP "wav-scp"
@@ -42,6 +45,11 @@
#define AM_CONFIG_NAME "config.yaml"
#define PUNC_CONFIG_NAME "punc.yaml"
#define ENCODER_NAME "model.onnx"
#define QUANT_ENCODER_NAME "model_quant.onnx"
#define DECODER_NAME "decoder.onnx"
#define QUANT_DECODER_NAME "decoder_quant.onnx"
// vad
#ifndef VAD_SILENCE_DURATION
#define VAD_SILENCE_DURATION 800
@@ -63,6 +71,19 @@
#define VAD_LFR_N 1
#endif
// asr
#ifndef PARA_LFR_M
#define PARA_LFR_M 7
#endif
#ifndef PARA_LFR_N
#define PARA_LFR_N 6
#endif
#ifndef ONLINE_STEP
#define ONLINE_STEP 9600
#endif
// punc
#define UNK_CHAR "<unk>"
#define TOKEN_LEN     20