tuna2134/sbv2-v2
1
0
Fork 0
Code Issues Pull Requests 1 Actions Packages Projects Releases Wiki Activity

Improve: Graphical display of ONNX conversion script logs

Browse Source
This commit is contained in:
tsukumi
2024-09-18 06:09:49 +09:00
parent 449927616a
commit 9b27ba9777
5 changed files with 74 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
style_bert_vits2/nlp/onnx_bert_models.py
View File

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

6
style_bert_vits2/tts_model.py
View File

@@ -132,7 +132,7 @@ class TTSModel:
hps=self.hyper_parameters,
)
logger.info(
f"Model loaded successfully from {self.model_path} to \"{self.device}\" device ({time.time() - start_time:.2f}s)"
f'Model loaded successfully from {self.model_path} to "{self.device}" device ({time.time() - start_time:.2f}s)'
)
# ONNX 推論時
@@ -140,7 +140,9 @@ class TTSModel:
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