ayo/ipex-llm
1
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions 1

Add trl version in error message (#13049)

Browse source 
* add version in error msg

* fix style
This commit is contained in:
Kai Huang 2025-05-15 09:16:27 +08:00 committed by GitHub
parent bd45bf7584
commit 35b49e4d91
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 2 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
python/llm/src/ipex_llm/transformers/speculative.py
View file

@ -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