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

多卡预处理

Browse Source
This commit is contained in:
Stardust·减
2023-09-05 19:35:21 +08:00
committed by GitHub
parent d02113267b
commit eb14f4a0b7
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

@@ -17,7 +17,7 @@ def cleaned_text_to_sequence(cleaned_text, tones, language):
lang_ids = [lang_id for i in phones] lang_ids = [lang_id for i in phones]
return phones, tones, lang_ids return phones, tones, lang_ids
def get_bert(norm_text, word2ph, language): 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 ja_bert
@@ -26,5 +26,5 @@ def get_bert(norm_text, word2ph, language):
'EN': en_bert, 'EN': en_bert,
'JA': ja_bert 'JA': ja_bert
} }
bert = lang_bert_func_map[language](norm_text, word2ph) bert = lang_bert_func_map[language](norm_text, word2ph, device)
return bert return bert