Improve: Graphical display of ONNX conversion script logs

2024-09-18 06:09:49 +09:00
parent 449927616a
commit 9b27ba9777
5 changed files with 74 additions and 14 deletions
--- a/server_editor.py
+++ b/server_editor.py
@@ -202,9 +202,7 @@ if args.preload_onnx_bert:
    )
    onnx_bert_models.load_tokenizer(Languages.JP)

-model_holder = TTSModelHolder(
-    model_dir, device, torch_device_to_onnx_providers(device)
-)
+model_holder = TTSModelHolder(model_dir, device, torch_device_to_onnx_providers(device))
 if len(model_holder.model_names) == 0:
    logger.error(f"Models not found in {model_dir}.")
    sys.exit(1)