Merge branch 'NVIDIA:main' into readd-swedish-itn

NVIDIA · Jul 16, 2023 · 28c1951 · 28c1951
2 parents bf6b0a3 + c7c549e
commit 28c1951
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/tools/text_processing_deployment/pynini_export.py b/tools/text_processing_deployment/pynini_export.py
@@ -191,6 +191,12 @@ def parse_args():
             VerbalizeFst as ITNVerbalizeFst,
         )
     elif args.language == 'zh':
+        from nemo_text_processing.inverse_text_normalization.zh.taggers.tokenize_and_classify import (
+            ClassifyFst as ITNClassifyFst,
+        )
+        from nemo_text_processing.inverse_text_normalization.zh.verbalizers.verbalize import (
+            VerbalizeFst as ITNVerbalizeFst,
+        )
         from nemo_text_processing.text_normalization.zh.taggers.tokenize_and_classify import (
             ClassifyFst as TNClassifyFst,
         )