We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 83cb0cb commit c885a54Copy full SHA for c885a54
optimum/exporters/openvino/model_patcher.py
@@ -15,9 +15,9 @@
15
import logging as log
16
17
from optimum.intel.utils.import_utils import (
18
- _openvino_version,
19
_torch_version,
20
_transformers_version,
+ is_openvino_version,
21
is_torch_version,
22
is_transformers_version,
23
)
@@ -41,7 +41,7 @@ def patch_model_with_bettertransformer(model):
41
+ COLOR_RESET
42
43
44
- if is_transformers_version(">=", "4.38") and _openvino_version < "2024.1.0-14612":
+ if is_transformers_version(">=", "4.38") and is_openvino_version("<", "2024.1.0-14612"):
45
log.warn(
46
COLOR_RED + f"[WARNING] Stateful models are not supported with Transformers {_transformers_version} and "
47
"this OpenVINO version. For good performance, consider using a nightly OpenVINO build: "
0 commit comments