add eng mock bert
This commit is contained in:
@@ -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
|
||||
|
||||
@@ -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__':
|
||||
|
||||
5
text/english_bert_mock.py
Normal file
5
text/english_bert_mock.py
Normal file
@@ -0,0 +1,5 @@
|
||||
import torch
|
||||
|
||||
|
||||
def get_bert_feature(norm_text, word2ph):
|
||||
return torch.zeros(1024, sum(word2ph))
|
||||
Reference in New Issue
Block a user