File tree Expand file tree Collapse file tree 1 file changed +9
-2
lines changed
onnxscript/_framework_apis Expand file tree Collapse file tree 1 file changed +9
-2
lines changed Original file line number Diff line number Diff line change 1414]
1515from typing import TYPE_CHECKING
1616
17- from onnxscript import ir , optimizer
17+ from onnxscript import ir , optimizer , version_converter
1818from onnxscript ._framework_apis .torch_2_5 import (
1919 check_model ,
20- convert_version ,
2120 get_torchlib_ops ,
2221 save_model_with_external_data ,
2322)
@@ -32,6 +31,14 @@ def optimize(model: ir.Model) -> ir.Model:
3231 return model
3332
3433
34+ def convert_version (model : ir .Model , target_version : int ) -> ir .Model :
35+ """Convert the model to the specified ONNX opset version."""
36+ if target_version < 18 :
37+ return model
38+ version_converter .convert_version (model , target_version )
39+ return model
40+
41+
3542def torchlib_opset () -> Opset18 :
3643 """Return the default opset for torchlib."""
3744 import onnxscript # pylint: disable=import-outside-toplevel
You can’t perform that action at this time.
0 commit comments