refine multilingual functionality

open-mmlab · Feb 17, 2024 · 7310116 · 7310116
1 parent 03df61e
commit 7310116
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/text/g2p_module.py b/text/g2p_module.py
@@ -147,6 +147,8 @@ def _initialize_backend(
                 words_mismatch=words_mismatch,
             )
         elif backend in ["pypinyin", "pypinyin_initials_finals"]:
+            if language != "cmn":
+                raise ValueError(f"{language} is not supported for pypinyin and pypinyin_initials_finals.")
             return PypinyinBackend(
                 backend=backend,
                 punctuation_marks=punctuation_marks + self.separator.word,