tuna2134/sbv2-v2
1
0
Fork 0
Code Issues Pull Requests 1 Actions Packages Projects Releases Wiki Activity

Refactor: moved text/ to style_bert_vits2/text_processing/(language)/

Browse Source
This commit is contained in:
tsukumi
2024-03-07 04:48:11 +00:00
parent d36401849b
commit f8f798d10a
10 changed files with 15 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
style_bert_vits2/text_processing/__init__.py
View File

@@ -67,18 +67,17 @@ def clean_text(
# Changed to import inside if condition to avoid unnecessary import
if language == Languages.JP:
from style_bert_vits2.text_processing.japanese.g2p import g2p
from style_bert_vits2.text_processing.japanese.normalizer import normalize_text
from style_bert_vits2.text_processing.japanese import g2p, normalize_text
norm_text = normalize_text(text)
phones, tones, word2ph = g2p(norm_text, use_jp_extra, raise_yomi_error)
elif language == Languages.EN:
from ...text import english as language_module
norm_text = language_module.normalize_text(text)
phones, tones, word2ph = language_module.g2p(norm_text)
from style_bert_vits2.text_processing.english import g2p, normalize_text
norm_text = normalize_text(text)
phones, tones, word2ph = g2p(norm_text)
elif language == Languages.ZH:
from ...text import chinese as language_module
norm_text = language_module.normalize_text(text)
phones, tones, word2ph = language_module.g2p(norm_text)
from style_bert_vits2.text_processing.chinese import g2p, normalize_text
norm_text = normalize_text(text)
phones, tones, word2ph = g2p(norm_text)
else:
raise ValueError(f"Language {language} not supported")