From 35b49e4d9115901cfc87ca708bf1985c308eb418 Mon Sep 17 00:00:00 2001 From: Kai Huang Date: Thu, 15 May 2025 09:16:27 +0800 Subject: [PATCH] Add trl version in error message (#13049) * add version in error msg * fix style --- python/llm/src/ipex_llm/transformers/speculative.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) 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