Fix: Disable optimization on InferenceSession initialization to speed up loading of ONNX models

This commit is contained in:
tsukumi
2024-09-22 23:08:00 +09:00
parent 16115a366b
commit 0e0b4bea31
4 changed files with 26 additions and 16 deletions

View File

@@ -21,7 +21,11 @@ from style_bert_vits2.nlp import bert_models
if __name__ == "__main__":
start_time = time.time()
parser = ArgumentParser()
parser.add_argument("--language", default=Languages.JP, help="Language of the BERT model to be converted")
parser.add_argument(
"--language",
default=Languages.JP,
help="Language of the BERT model to be converted",
)
args = parser.parse_args()
# モデルの入出力先ファイルパスを取得