fix(wenetruntime): add ITN for python bingding (#2042)

wenet-e2e · Oct 11, 2023 · 32c9c00 · 32c9c00
1 parent 35acffc
commit 32c9c00
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/runtime/binding/python/CMakeLists.txt b/runtime/binding/python/CMakeLists.txt
@@ -4,6 +4,8 @@ project(wenet VERSION 0.1)
 
 option(TORCH "whether to build with Torch" ON)
 option(ONNX "whether to build with ONNX" OFF)
+option(ITN "whether to use WeTextProcessing" ON)
+
 set(CXX11_ABI OFF)
 set(FST_HAVE_BIN OFF)
 set(CMAKE_VERBOSE_MAKEFILE OFF)
@@ -39,6 +41,10 @@ include_directories(
   ${CMAKE_CURRENT_SOURCE_DIR}/kaldi
 )
 
+if(ITN)
+  include(wetextprocessing)
+endif()
+
 add_subdirectory(utils)
 add_subdirectory(frontend)
 add_subdirectory(post_processor)