fix ja bert path

2024-02-27 17:18:20 +00:00
parent 63869183f8
commit 29f9c65819
2 changed files with 3 additions and 7 deletions
--- a/melo/text/japanese_bert.py
+++ b/melo/text/japanese_bert.py
@@ -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