Fix: PyTorch inference test

This commit is contained in:
tsukumi
2024-09-23 15:20:52 +09:00
parent e91b69862c
commit 08af692835

View File

@@ -47,8 +47,9 @@ def synthesize(
# ロードされた InferenceSession の ExecutionProvider が一致するか確認 # ロードされた InferenceSession の ExecutionProvider が一致するか確認
# 一致しない場合、指定された ExecutionProvider で推論できない状態 # 一致しない場合、指定された ExecutionProvider で推論できない状態
assert model.onnx_session is not None if inference_type == "onnx":
assert model.onnx_session.get_providers()[0] == onnx_providers[0][0] assert model.onnx_session is not None
assert model.onnx_session.get_providers()[0] == onnx_providers[0][0]
# すべてのスタイルに対して音声合成を実行 # すべてのスタイルに対して音声合成を実行
for style in model_info.styles: for style in model_info.styles: