From e1d9fd53f0cbd81410a3a24d243188642d6331e4 Mon Sep 17 00:00:00 2001 From: "Liu, Kaixuan" Date: Tue, 17 Dec 2024 12:51:59 -0500 Subject: [PATCH] fix import bug when ipex is available but sentence_transformers is not; add `accelerate` dependency to run with XPU Signed-off-by: Liu, Kaixuan --- optimum/intel/__init__.py | 1 + setup.py | 2 +- 2 files changed, 2 insertions(+), 1 deletion(-) diff --git a/optimum/intel/__init__.py b/optimum/intel/__init__.py index 335eed5638..b441b76f93 100644 --- a/optimum/intel/__init__.py +++ b/optimum/intel/__init__.py @@ -51,6 +51,7 @@ "IPEXModel", ] else: + _import_structure["utils.dummy_ipex_objects"] = [] _import_structure["ipex"] = [ "IPEXModelForCausalLM", "IPEXModelForSequenceClassification", diff --git a/setup.py b/setup.py index ca415fca35..d9b3b8642b 100644 --- a/setup.py +++ b/setup.py @@ -66,7 +66,7 @@ "nncf": ["nncf>=2.14.0"], "openvino": ["nncf>=2.14.0", "openvino>=2024.5.0", "openvino-tokenizers>=2024.5.0"], "neural-compressor": ["neural-compressor[pt]>3.0", "accelerate", "transformers<4.46"], - "ipex": ["intel-extension-for-pytorch>=2.4", "transformers>4.45,<4.47"], + "ipex": ["intel-extension-for-pytorch>=2.4", "transformers>4.45,<4.47", "accelerate"], "diffusers": ["diffusers"], "quality": QUALITY_REQUIRE, "tests": TESTS_REQUIRE,