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

Improve to import only necesary lang

Browse Source
This commit is contained in:
litagin02
2024-02-23 16:20:08 +09:00
parent 0a6e4f1203
commit cb842faaf7
1 changed files with 17 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

27
text/__init__.py
View File

@@ -27,17 +27,24 @@ def get_bert(
assist_text_weight=0.7,
ignore_unknown=False,
):
from .chinese_bert import get_bert_feature as zh_bert
from .english_bert_mock import get_bert_feature as en_bert
from .japanese_bert import get_bert_feature as jp_bert
# from .chinese_bert import get_bert_feature as zh_bert
# from .english_bert_mock import get_bert_feature as en_bert
# from .japanese_bert import get_bert_feature as jp_bert
lang_bert_func_map = {"ZH": zh_bert, "EN": en_bert, "JP": jp_bert}
if language == "JP":
bert = lang_bert_func_map[language](
# Change logic to import inside if condition to avoid unnecessary import
if language == "ZH":
from .chinese_bert import get_bert_feature as zh_bert
return zh_bert(text, word2ph, device, assist_text, assist_text_weight)
elif language == "EN":
from .english_bert_mock import get_bert_feature as en_bert
return en_bert(text, word2ph, device, assist_text, assist_text_weight)
elif language == "JP":
from .japanese_bert import get_bert_feature as jp_bert
return jp_bert(
text, word2ph, device, assist_text, assist_text_weight, ignore_unknown
)
else:
bert = lang_bert_func_map[language](
text, word2ph, device, assist_text, assist_text_weight
)
return bert
raise ValueError(f"Language {language} not supported")