From d9c808a8df49a747b484993f7b69dc9c4936992b Mon Sep 17 00:00:00 2001
From: 北念 <lzr265946@alibaba-inc.com>
Date: 星期四, 12 十月 2023 17:13:20 +0800
Subject: [PATCH] update setup.py

---
 funasr/runtime/onnxruntime/include/com-define.h |    4 ++++
 1 files changed, 4 insertions(+), 0 deletions(-)

diff --git a/funasr/runtime/onnxruntime/include/com-define.h b/funasr/runtime/onnxruntime/include/com-define.h
index e08d50f..9f28e15 100644
--- a/funasr/runtime/onnxruntime/include/com-define.h
+++ b/funasr/runtime/onnxruntime/include/com-define.h
@@ -50,6 +50,10 @@
 #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"
 #define DECODER_NAME "decoder.onnx"

--
Gitblit v1.9.1