Skip to content

Commit 49d297c

Browse files
committed
fallback C.simplify ir_version error to simplify_path
1 parent fbf1ca8 commit 49d297c

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

onnxsim/onnx_simplifier.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -209,7 +209,7 @@ def parse_size(size: str) -> int:
209209
check_ok = model_checking.compare(
210210
model_opt, model, check_n, test_input_shapes, input_data, custom_lib
211211
)
212-
except ValueError:
212+
except (ValueError, onnx.onnx_cpp2py_export.checker.ValidationError):
213213
print("[bold magenta]Simplified model larger than 2GB. Trying to save as external data...[/bold magenta]")
214214
# large models try to convert through a temporary file
215215
with tempfile.TemporaryDirectory() as tmpdirname:

0 commit comments

Comments
 (0)