add eng mock bert

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

View File

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

View File

@@ -22,7 +22,7 @@ def clean_text_bert(text, language):
return phones, tones, bert
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)
if __name__ == '__main__':

View File

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