File tree 1 file changed +9
-2
lines changed
onnxscript/_framework_apis
1 file changed +9
-2
lines changed Original file line number Diff line number Diff line change 14
14
]
15
15
from typing import TYPE_CHECKING
16
16
17
- from onnxscript import ir , optimizer
17
+ from onnxscript import ir , optimizer , version_converter
18
18
from onnxscript ._framework_apis .torch_2_5 import (
19
19
check_model ,
20
- convert_version ,
21
20
get_torchlib_ops ,
22
21
save_model_with_external_data ,
23
22
)
@@ -32,6 +31,14 @@ def optimize(model: ir.Model) -> ir.Model:
32
31
return model
33
32
34
33
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
+
35
42
def torchlib_opset () -> Opset18 :
36
43
"""Return the default opset for torchlib."""
37
44
import onnxscript # pylint: disable=import-outside-toplevel
You can’t perform that action at this time.
0 commit comments