We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 9bd1baa commit 544aa18Copy full SHA for 544aa18
modelopt/onnx/trt_utils.py
@@ -332,6 +332,9 @@ def load_onnx_model(
332
save_onnx(onnx_model, ir_version_onnx_path, use_external_data_format)
333
intermediate_generated_files.append(ir_version_onnx_path) # type: ignore[union-attr]
334
335
+ # Check that the model is valid
336
+ onnx.checker.check_model(onnx_model)
337
+
338
return (
339
onnx_model,
340
has_custom_op,
0 commit comments