| import importlib |
| from bert_vits2.text import cleaned_text_to_sequence |
|
|
| language_module_map = { |
| 'zh': "bert_vits2.text.chinese", |
| 'ja': "bert_vits2.text.japanese" |
| } |
|
|
| _loaded_modules = {} |
|
|
|
|
| def get_language_module(language): |
| if language not in _loaded_modules: |
| module_path = language_module_map.get(language) |
| if not module_path: |
| raise ValueError(f"Unsupported language: {language}") |
|
|
| _loaded_modules[language] = importlib.import_module(module_path) |
|
|
| return _loaded_modules[language] |
|
|
|
|
| def clean_text(text, language): |
| language_module = get_language_module(language) |
| norm_text = language_module.text_normalize(text) |
| phones, tones, word2ph = language_module.g2p(norm_text) |
| return norm_text, phones, tones, word2ph |
|
|
|
|
| def clean_text_bert(text, language): |
| language_module = get_language_module(language) |
| norm_text = language_module.text_normalize(text) |
| phones, tones, word2ph = language_module.g2p(norm_text) |
| bert = language_module.get_bert_feature(norm_text, word2ph) |
| return phones, tones, bert |
|
|
|
|
| def text_to_sequence(text, language): |
| norm_text, phones, tones, word2ph = clean_text(text, language) |
| return cleaned_text_to_sequence(phones, tones, language) |
|
|
|
|
| if __name__ == '__main__': |
| pass |
|
|
