support WeTextProcessing

wenet-e2e · Nov 8, 2023 · da2b532 · da2b532
1 parent 8cdaa76
commit da2b532
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/runtime/openvino/CMakeLists.txt b/runtime/openvino/CMakeLists.txt
@@ -15,6 +15,7 @@ option(TORCH "whether to build with Torch" OFF)
 option(ONNX "whether to build with ONNX" OFF)
 option(GPU "whether to build with GPU" OFF)
 option(OPENVINO "whether to build with OpenVINO" ON)
+option(ITN "whether to use WeTextProcessing" ON)
 
 set(CMAKE_VERBOSE_MAKEFILE OFF)
 
@@ -41,6 +42,10 @@ include_directories(
   ${CMAKE_CURRENT_SOURCE_DIR}/kaldi
 )
 
+if(ITN)
+  include(wetextprocessing)
+endif()
+
 # Build all libraries
 add_subdirectory(utils)
 add_subdirectory(frontend)