Skip to content

Commit

Permalink
fix for runing RunONNXModel without verify (#2876)
Browse files Browse the repository at this point in the history
Signed-off-by: Alexandre Eichenberger <alexe@us.ibm.com>
  • Loading branch information
AlexandreEichenberger authored Jul 13, 2024
1 parent 94d9a8c commit a198252
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion utils/RunONNXModel.py
Original file line number Diff line number Diff line change
Expand Up @@ -231,7 +231,7 @@ def check_non_negative(argname, value):
"executables and libraries can be found, typically `onnx-mlir/build/Debug`"
)

if args.verify.lower() == "onnxruntime":
if args.verify and args.verify.lower() == "onnxruntime":
if not args.model or (args.model and not args.model.endswith(".onnx")):
raise RuntimeError(
"Set input onnx model using argument --model when verifying using onnxruntime."
Expand Down

0 comments on commit a198252

Please sign in to comment.