Skip to content

Commit 01ef856

Browse files
committed
comments
1 parent 3d1f428 commit 01ef856

File tree

1 file changed

+8
-2
lines changed

1 file changed

+8
-2
lines changed

onnxscript/version_converter/__init__.py

Lines changed: 8 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -87,15 +87,21 @@ def call(self, model: ir.Model) -> ir.passes.PassResult:
8787
# No need to convert the version
8888
return ir.passes.PassResult(model, False)
8989

90-
if _version_converter.version_supported(model, self.target_version):
90+
# When fallback is disabled, always use the onnxscript version converter;
91+
# When fallback is enabled, use the onnxscript version converter
92+
# if the target version is supported. Otherwise, use the onnx C API
93+
# to convert the model.
94+
if not self.fallback or _version_converter.version_supported(
95+
model, self.target_version
96+
):
9197
_version_converter.convert_version(
9298
model,
9399
target_version=self.target_version,
94100
)
95101
return ir.passes.PassResult(model, True)
96102

97103
if not self.fallback:
98-
logger.info(
104+
logger.warning(
99105
"The model version conversion is not supported by the onnxscript version converter "
100106
"and fallback is disabled. The model was not modified"
101107
" (target version: %d). "

0 commit comments

Comments
 (0)