File tree 1 file changed +4
-0
lines changed
voicevox_engine/tts_pipeline
1 file changed +4
-0
lines changed Original file line number Diff line number Diff line change @@ -149,18 +149,22 @@ def __init__(
149
149
# 音声合成に必要な BERT モデル・トークナイザーを読み込む
150
150
## 最新のモデルがまだローカルにキャッシュされていない場合は、自動的にネットワークからダウンロードされる
151
151
## 一度ロードしておけば、同じプロセス内でグローバルに保持される
152
+ ## リビジョンを指定しない場合毎回 Hugging Face への通信が発生し、オフライン環境では 60 秒でタイムアウトするまで待たされるので、
153
+ ## 明示的にコミットハッシュでリビジョンを指定している (こうすることで、オンライン環境でもロード時間が短縮されるメリットもある)
152
154
start_time = time .time ()
153
155
logger .info ("Loading BERT model and tokenizer..." )
154
156
onnx_bert_models .load_model (
155
157
language = Languages .JP ,
156
158
pretrained_model_name_or_path = "tsukumijima/deberta-v2-large-japanese-char-wwm-onnx" ,
157
159
onnx_providers = self .onnx_providers ,
158
160
cache_dir = str (self .BERT_MODEL_CACHE_DIR ),
161
+ revision = "d701ec67708287b20d2063270f6b535e6eed09ab" ,
159
162
)
160
163
onnx_bert_models .load_tokenizer (
161
164
language = Languages .JP ,
162
165
pretrained_model_name_or_path = "tsukumijima/deberta-v2-large-japanese-char-wwm-onnx" ,
163
166
cache_dir = str (self .BERT_MODEL_CACHE_DIR ),
167
+ revision = "d701ec67708287b20d2063270f6b535e6eed09ab" ,
164
168
)
165
169
logger .info (
166
170
f"BERT model and tokenizer loaded. ({ time .time () - start_time :.2f} s)"
You can’t perform that action at this time.
0 commit comments