Improve: Add option to generate AIVM/AIVMX files directly when running convert_onnx.py, add license information

2025-03-26 20:01:17 +09:00
parent 810ca43615
commit 2aa503233a
3 changed files with 396 additions and 255 deletions
--- a/style_bert_vits2/nlp/onnx_bert_models.py
+++ b/style_bert_vits2/nlp/onnx_bert_models.py
@@ -106,9 +106,7 @@ def load_model(
    # 推論時に一番優先される ExecutionProvider の名前を取得
    assert len(onnx_providers) > 0
    first_provider_name = (
-        onnx_providers[0]
-        if type(onnx_providers[0]) is str
-        else onnx_providers[0][0]
+        onnx_providers[0] if type(onnx_providers[0]) is str else onnx_providers[0][0]
    )

    # 推論セッションの設定