diff --git a/python/llm/src/ipex_llm/transformers/speculative.py b/python/llm/src/ipex_llm/transformers/speculative.py index d0c90b10..b7e486ed 100644 --- a/python/llm/src/ipex_llm/transformers/speculative.py +++ b/python/llm/src/ipex_llm/transformers/speculative.py @@ -54,7 +54,8 @@ if version.parse(trans_version) >= version.parse("4.39.0"): try: from trl.core import top_k_top_p_filtering except ModuleNotFoundError: - log4Error.invalidInputError(False, "For transformers version >= 4.39.0, pip install trl") + log4Error.invalidInputError(False, + "For transformers version >= 4.39.0, pip install trl==0.11.0") else: from transformers import top_k_top_p_filtering