Skip to content

Commit c885a54

Browse files
committed
Use is_openvino_version to compare versions
1 parent 83cb0cb commit c885a54

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

optimum/exporters/openvino/model_patcher.py

+2-2
Original file line numberDiff line numberDiff line change
@@ -15,9 +15,9 @@
1515
import logging as log
1616

1717
from optimum.intel.utils.import_utils import (
18-
_openvino_version,
1918
_torch_version,
2019
_transformers_version,
20+
is_openvino_version,
2121
is_torch_version,
2222
is_transformers_version,
2323
)
@@ -41,7 +41,7 @@ def patch_model_with_bettertransformer(model):
4141
+ COLOR_RESET
4242
)
4343

44-
if is_transformers_version(">=", "4.38") and _openvino_version < "2024.1.0-14612":
44+
if is_transformers_version(">=", "4.38") and is_openvino_version("<", "2024.1.0-14612"):
4545
log.warn(
4646
COLOR_RED + f"[WARNING] Stateful models are not supported with Transformers {_transformers_version} and "
4747
"this OpenVINO version. For good performance, consider using a nightly OpenVINO build: "

0 commit comments

Comments
 (0)