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

fix ipex_llm import in transformers 4.45 (#12099)

Browse source
This commit is contained in:
Yishuo Wang 2024-09-20 15:24:59 +08:00 committed by GitHub
parent a6cbc01911
commit 54b973c744
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 3 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
python/llm/src/ipex_llm/utils/__init__.py
View file

@ -22,7 +22,9 @@ import transformers
trans_version = transformers.__version__ trans_version = transformers.__version__
if trans_version >= "4.43.0": if trans_version >= "4.45.0":
pass
elif trans_version >= "4.43.0":
from .benchmark_util_4_43 import BenchmarkWrapper from .benchmark_util_4_43 import BenchmarkWrapper
elif trans_version >= "4.42.0": elif trans_version >= "4.42.0":
from .benchmark_util_4_42 import BenchmarkWrapper from .benchmark_util_4_42 import BenchmarkWrapper