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

add eng mock bert

Browse Source
This commit is contained in:
rcell
2023-07-21 13:34:44 +08:00
parent b601991286
commit abac0cb146
3 changed files with 9 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
text/__init__.py
View File

@@ -19,8 +19,10 @@ def cleaned_text_to_sequence(cleaned_text, tones, language):
def get_bert(norm_text, word2ph, language): def get_bert(norm_text, word2ph, language):
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
lang_bert_func_map = { lang_bert_func_map = {
'ZH': zh_bert 'ZH': zh_bert,
'EN': en_bert
} }
bert = lang_bert_func_map[language](norm_text, word2ph) bert = lang_bert_func_map[language](norm_text, word2ph)
return bert return bert

2
text/cleaner.py
View File

@@ -22,7 +22,7 @@ def clean_text_bert(text, language):
return phones, tones, bert return phones, tones, bert
def text_to_sequence(text, language): def text_to_sequence(text, language):
_, _, phones, tones = clean_text(text, language) norm_text, phones, tones, word2ph = clean_text(text, language)
return cleaned_text_to_sequence(phones, tones, language) return cleaned_text_to_sequence(phones, tones, language)
if __name__ == '__main__': if __name__ == '__main__':

5
text/english_bert_mock.py Normal file
View File

@@ -0,0 +1,5 @@
import torch
def get_bert_feature(norm_text, word2ph):
return torch.zeros(1024, sum(word2ph))