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)
|