From 17816181af46a4e34abe01d465b669a863fde195 Mon Sep 17 00:00:00 2001 From: Kai Huang Date: Mon, 7 Apr 2025 11:31:02 +0800 Subject: [PATCH 1/2] add version in error msg --- python/llm/src/ipex_llm/transformers/speculative.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/python/llm/src/ipex_llm/transformers/speculative.py b/python/llm/src/ipex_llm/transformers/speculative.py index d0c90b1053a..f9622fcb9b2 100644 --- a/python/llm/src/ipex_llm/transformers/speculative.py +++ b/python/llm/src/ipex_llm/transformers/speculative.py @@ -54,7 +54,7 @@ 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 From ad6cbd152271916d176e4479087b8be2b9a3df01 Mon Sep 17 00:00:00 2001 From: Kai Huang Date: Mon, 7 Apr 2025 11:32:43 +0800 Subject: [PATCH 2/2] 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 f9622fcb9b2..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==0.11.0") + log4Error.invalidInputError(False, + "For transformers version >= 4.39.0, pip install trl==0.11.0") else: from transformers import top_k_top_p_filtering