zhangshuai/MeloTTS
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix ja bert path

Browse Source
This commit is contained in:
qinzy
2024-02-27 17:18:20 +00:00
parent 63869183f8
commit 29f9c65819
2 changed files with 3 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
melo/text/japanese.py
View File

@@ -566,7 +566,7 @@ def distribute_phone(n_phone, n_word):
# tokenizer = AutoTokenizer.from_pretrained('cl-tohoku/bert-base-japanese-v3')
model_id = 'cl-tohoku/bert-base-japanese-v3'
model_id = 'tohoku-nlp/bert-base-japanese-v3'
tokenizer = AutoTokenizer.from_pretrained(model_id)
def g2p(norm_text):

6
melo/text/japanese_bert.py
View File

@@ -3,13 +3,9 @@ from transformers import AutoTokenizer, AutoModelForMaskedLM
import sys
# model = None
# model_id = 'cl-tohoku/bert-base-japanese-v3'
# tokenizer = AutoTokenizer.from_pretrained(model_id)
models = {}
tokenizers = {}
def get_bert_feature(text, word2ph, device=None, model_id='cl-tohoku/bert-base-japanese-v3'):
def get_bert_feature(text, word2ph, device=None, model_id='tohoku-nlp/bert-base-japanese-v3'):
global model
global tokenizer