From 8f42ca2df4bd5eb379895b3e903163c888ccd1dd Mon Sep 17 00:00:00 2001 From: Yishuo Wang Date: Fri, 20 Sep 2024 14:19:16 +0800 Subject: [PATCH] fix ipex_llm import in transformers 4.45 --- python/llm/src/ipex_llm/utils/__init__.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/python/llm/src/ipex_llm/utils/__init__.py b/python/llm/src/ipex_llm/utils/__init__.py index 9490597d89a..1d2c80c6f50 100644 --- a/python/llm/src/ipex_llm/utils/__init__.py +++ b/python/llm/src/ipex_llm/utils/__init__.py @@ -22,7 +22,9 @@ 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 elif trans_version >= "4.42.0": from .benchmark_util_4_42 import BenchmarkWrapper