fix import bug when ipex is available but sentence_transformers is not; add accelerate dependency to run with XPU (#1077)

kaixuanliu · web-flow · commit 7601bfd43645 · 2024-12-17T10:53:36.000+01:00
Signed-off-by: Liu, Kaixuan &lt;kaixuan.liu@intel.com&gt;
diff --git 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
@@ -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,

Original file line number	Diff line number	Diff line change
`@@ -51,6 +51,7 @@`
`51`	`51`	`"IPEXModel",`
`52`	`52`	`]`
`53`	`53`	`else:`
	`54`	`+ _import_structure["utils.dummy_ipex_objects"] = []`
`54`	`55`	`_import_structure["ipex"] = [`
`55`	`56`	`"IPEXModelForCausalLM",`
`56`	`57`	`"IPEXModelForSequenceClassification",`