update server_editor,py
Browse files- server_editor.py +2 -2
server_editor.py
CHANGED
@@ -54,7 +54,7 @@ from style_bert_vits2.nlp.japanese.user_dict import (
|
|
54 |
update_dict,
|
55 |
)
|
56 |
from style_bert_vits2.tts_model import TTSModelHolder, TTSModelInfo
|
57 |
-
|
58 |
|
59 |
# ---フロントエンド部分に関する処理---
|
60 |
|
@@ -195,7 +195,7 @@ if not args.skip_default_models:
|
|
195 |
download_default_models()
|
196 |
skip_static_files = bool(args.skip_static_files)
|
197 |
|
198 |
-
model_holder = TTSModelHolder(model_dir, device)
|
199 |
if len(model_holder.model_names) == 0:
|
200 |
logger.error(f"Models not found in {model_dir}.")
|
201 |
sys.exit(1)
|
|
|
54 |
update_dict,
|
55 |
)
|
56 |
from style_bert_vits2.tts_model import TTSModelHolder, TTSModelInfo
|
57 |
+
from style_bert_vits2.utils import torch_device_to_onnx_providers
|
58 |
|
59 |
# ---フロントエンド部分に関する処理---
|
60 |
|
|
|
195 |
download_default_models()
|
196 |
skip_static_files = bool(args.skip_static_files)
|
197 |
|
198 |
+
model_holder = TTSModelHolder(model_dir, device, torch_device_to_onnx_providers(device))
|
199 |
if len(model_holder.model_names) == 0:
|
200 |
logger.error(f"Models not found in {model_dir}.")
|
201 |
sys.exit(1)
|