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

Update __init__.py

Browse Source
This commit is contained in:
Stardust·减
2023-09-05 21:53:37 +08:00
committed by GitHub
parent edda628161
commit a4ea278bc3
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
text/__init__.py
View File

@@ -20,11 +20,11 @@ def cleaned_text_to_sequence(cleaned_text, tones, language):
def get_bert(norm_text, word2ph, language, device): def get_bert(norm_text, word2ph, language, device):
from .chinese_bert import get_bert_feature as zh_bert from .chinese_bert import get_bert_feature as zh_bert
from .english_bert_mock import get_bert_feature as en_bert from .english_bert_mock import get_bert_feature as en_bert
from .japanese_bert import get_bert_feature as ja_bert from .japanese_bert import get_bert_feature as jp_bert
lang_bert_func_map = { lang_bert_func_map = {
'ZH': zh_bert, 'ZH': zh_bert,
'EN': en_bert, 'EN': en_bert,
'JA': ja_bert 'JP': jp_bert
} }
bert = lang_bert_func_map[language](norm_text, word2ph, device) bert = lang_bert_func_map[language](norm_text, word2ph, device)
return bert return bert