KO/ZH model soon

2024-12-15 17:01:14 -06:00 · 2024-12-15 17:01:14 -06:00 · cd4a5f427c
commit cd4a5f427c
parent 4800e7179a
1 changed files with 2 additions and 1 deletions
--- a/vall_e/emb/g2p.py
+++ b/vall_e/emb/g2p.py
@ -50,7 +50,8 @@ def coerce_language( lang ):
 		return "fr-fr"
 	# phonemizer/espeak used to have zh refer to mandarin, but was renamed to cmn
 	# cmn outputs cringe, but not cmn-latn-pinyin
-	if lang == "zh":
+	# also just coerces any of the dialects into this (to avoid crimes)
+	if lang[:2] == "zh":
 		return "cmn-latn-pinyin"
 	"""
 	things to consider in the future