diff --git a/python/llm/src/ipex_llm/transformers/speculative.py b/python/llm/src/ipex_llm/transformers/speculative.py index d0c90b1053a..b7e486ed4ce 100644 --- a/python/llm/src/ipex_llm/transformers/speculative.py +++ b/python/llm/src/ipex_llm/transformers/speculative.py @@ -54,7 +54,8 @@ 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