saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-05-25 22:26:24 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Fix onnx version comparison (#12223)

Browse source 
use version.parse to compare version
This commit is contained in:
Tianlei Wu 2022-07-20 11:14:06 -07:00 committed by GitHub
parent 43e1e89453
commit 5651d91c32
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
onnxruntime/python/tools/transformers/float16.py
View file

@ -14,6 +14,7 @@ import numpy as np
import onnx
from onnx import helper, numpy_helper
from onnx import onnx_pb as onnx_proto
from packaging import version
logger = logging.getLogger(__name__)
@ -170,7 +171,7 @@ def convert_float_to_float16(
assert max_finite_val <= float(np.finfo(np.float16).max), "invalid max_finite_val. largest float16 value: 65504"
func_infer_shape = None
if not disable_shape_infer and onnx.__version__ >= "1.2":
if not disable_shape_infer and version.parse(onnx.__version__) >= version.parse("1.2.0"):
try:
from onnx.shape_inference import infer_shapes