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,6 +47,7 @@ def synthesize(
# ロードされた InferenceSession の ExecutionProvider が一致するか確認
# 一致しない場合、指定された ExecutionProvider で推論できない状態
if inference_type == "onnx":
assert model.onnx_session is not None
assert model.onnx_session.get_providers()[0] == onnx_providers[0][0]