Skip to content

Commit 004135b

Browse files
committed
Include affected model types in warning message
1 parent dea1462 commit 004135b

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
@@ -52,8 +52,8 @@ def patch_model_with_bettertransformer(model):
5252
_openvino_version.split("-")[0] if is_openvino_version("<=", "2024.0.0-14509") else _openvino_version
5353
)
5454
log.warn(
55-
COLOR_RED + f"[WARNING] Stateful models are not supported with Transformers {_transformers_version} and "
56-
f"OpenVINO {display_version}. For good performance, consider using a nightly OpenVINO build: "
55+
COLOR_RED + f"[WARNING] Stateful models are not supported for Llama and GPTBigCode with Transformers "
56+
f"{_transformers_version} and OpenVINO {display_version}. For good performance, consider using a nightly OpenVINO build: "
5757
"https://docs.openvino.ai/2024/get-started/install-openvino.html. For models that do not need transformers "
5858
"4.38+, it is also an option to downgrade transformers: `pip install transformers==4.37.2`" + COLOR_RESET
5959
)

0 commit comments

Comments
 (0)