From 54b973c744c00251f66d376d985d244fcc03aea6 Mon Sep 17 00:00:00 2001 From: Yishuo Wang Date: Fri, 20 Sep 2024 15:24:59 +0800 Subject: [PATCH] fix ipex_llm import in transformers 4.45 (#12099) --- 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 9490597d..1d2c80c6 100644 --- a/python/llm/src/ipex_llm/utils/__init__.py +++ b/python/llm/src/ipex_llm/utils/__init__.py @@ -22,7 +22,9 @@ import transformers 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