bert読み込みを修正
Browse files- server_editor.py +3 -3
server_editor.py
CHANGED
|
@@ -42,7 +42,7 @@ from style_bert_vits2.constants import (
|
|
| 42 |
Languages,
|
| 43 |
)
|
| 44 |
from style_bert_vits2.logging import logger
|
| 45 |
-
from style_bert_vits2.nlp import
|
| 46 |
from style_bert_vits2.nlp.japanese import pyopenjtalk_worker as pyopenjtalk
|
| 47 |
from style_bert_vits2.nlp.japanese.g2p_utils import g2kata_tone, kata_tone2phone_tone
|
| 48 |
from style_bert_vits2.nlp.japanese.normalizer import normalize_text
|
|
@@ -160,8 +160,8 @@ update_dict()
|
|
| 160 |
# 事前に BERT モデル/トークナイザーをロードしておく
|
| 161 |
## ここでロードしなくても必要になった際に自動ロードされるが、時間がかかるため事前にロードしておいた方が体験が良い
|
| 162 |
## server_editor.py は日本語にしか対応していないため、日本語の BERT モデル/トークナイザーのみロードする
|
| 163 |
-
|
| 164 |
-
|
| 165 |
|
| 166 |
|
| 167 |
class AudioResponse(Response):
|
|
|
|
| 42 |
Languages,
|
| 43 |
)
|
| 44 |
from style_bert_vits2.logging import logger
|
| 45 |
+
from style_bert_vits2.nlp import onnx_bert_models
|
| 46 |
from style_bert_vits2.nlp.japanese import pyopenjtalk_worker as pyopenjtalk
|
| 47 |
from style_bert_vits2.nlp.japanese.g2p_utils import g2kata_tone, kata_tone2phone_tone
|
| 48 |
from style_bert_vits2.nlp.japanese.normalizer import normalize_text
|
|
|
|
| 160 |
# 事前に BERT モデル/トークナイザーをロードしておく
|
| 161 |
## ここでロードしなくても必要になった際に自動ロードされるが、時間がかかるため事前にロードしておいた方が体験が良い
|
| 162 |
## server_editor.py は日本語にしか対応していないため、日本語の BERT モデル/トークナイザーのみロードする
|
| 163 |
+
onnx_bert_models.load_model(Languages.JP)
|
| 164 |
+
onnx_bert_models.load_tokenizer(Languages.JP)
|
| 165 |
|
| 166 |
|
| 167 |
class AudioResponse(Response):
|