Merge pull request #336 from triple-Mu/triplemu/fix-ir-version-bug

daquexian · web-flow · commit 1379b1422e58 · 2024-07-08T13:57:01.000+01:00
fallback C.simplify ir_version error to C.simplify_path
diff --git a/onnxsim/onnx_simplifier.py b/onnxsim/onnx_simplifier.py
@@ -209,7 +209,7 @@ def parse_size(size: str) -> int:
         check_ok = model_checking.compare(
             model_opt, model, check_n, test_input_shapes, input_data, custom_lib
         )
-    except ValueError:
+    except (ValueError, onnx.onnx_cpp2py_export.checker.ValidationError):
         print("[bold magenta]Simplified model larger than 2GB. Trying to save as external data...[/bold magenta]")
         # large models try to convert through a temporary file
         with tempfile.TemporaryDirectory() as tmpdirname:

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