WariHima commited on
Commit
ab9b312
·
1 Parent(s): f2010a7

update server_editor,py

Browse files
Files changed (1) hide show
  1. 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)