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/style_bert_vits2/nlp/onnx_bert_models.py
+++ b/style_bert_vits2/nlp/onnx_bert_models.py
@@ -97,7 +97,9 @@ def load_model(
    sess_options = onnxruntime.SessionOptions()
    # 基本的な最適化のみ有効化
    # ONNX モデルの作成時にすでに onnxsim により最適化されているため、ここでは基本的な最適化のみ有効化する
-    sess_options.graph_optimization_level = onnxruntime.GraphOptimizationLevel.ORT_ENABLE_BASIC
+    sess_options.graph_optimization_level = (
+        onnxruntime.GraphOptimizationLevel.ORT_ENABLE_BASIC
+    )
    # エラー以外のログを出力しない
    # 本来は log_severity_level = 3 だけで効くはずだが、なぜか抑制できないので set_default_logger_severity() も呼び出している
    sess_options.log_severity_level = 3