From abac0cb14635352d725174c49d0d819865ea34a1 Mon Sep 17 00:00:00 2001 From: rcell Date: Fri, 21 Jul 2023 13:34:44 +0800 Subject: [PATCH] add eng mock bert --- text/__init__.py | 4 +++- text/cleaner.py | 2 +- text/english_bert_mock.py | 5 +++++ 3 files changed, 9 insertions(+), 2 deletions(-) create mode 100644 text/english_bert_mock.py diff --git a/text/__init__.py b/text/__init__.py index 8dcf065..1e22c4a 100644 --- a/text/__init__.py +++ b/text/__init__.py @@ -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 diff --git a/text/cleaner.py b/text/cleaner.py index 475f5f0..3b6a720 100644 --- a/text/cleaner.py +++ b/text/cleaner.py @@ -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__': diff --git a/text/english_bert_mock.py b/text/english_bert_mock.py new file mode 100644 index 0000000..3b894ce --- /dev/null +++ b/text/english_bert_mock.py @@ -0,0 +1,5 @@ +import torch + + +def get_bert_feature(norm_text, word2ph): + return torch.zeros(1024, sum(word2ph))