change ipex transformers limited verison in setup (#1015)

* change ipex transformers limited verison in setup * fix inc tests Signed-off-by: jiqing-feng <[email protected]>
huggingface · Nov 22, 2024 · a5c48a8 · a5c48a8
1 parent ed35ffc
commit a5c48a8
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 5 deletions.
diff --git a/setup.py b/setup.py
@@ -29,7 +29,7 @@
 INSTALL_REQUIRE = [
     "torch>=1.11",
     "optimum~=1.23",
-    "transformers>=4.36,<4.46",
+    "transformers>=4.36,<4.47",
     "datasets>=1.4.0",
     "sentencepiece",
     "setuptools",
@@ -62,9 +62,9 @@
 
 EXTRAS_REQUIRE = {
     "nncf": ["nncf>=2.11.0"],
-    "ipex": ["intel-extension-for-pytorch", "transformers>=4.39,<4.46"],
     "openvino": ["nncf>=2.11.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", "transformers>=4.39,<4.46"],
     "diffusers": ["diffusers"],
     "quality": QUALITY_REQUIRE,
     "tests": TESTS_REQUIRE,

diff --git a/tests/neural_compressor/test_cli.py b/tests/neural_compressor/test_cli.py
@@ -12,7 +12,6 @@ def test_string_type(self):
         self.assertIsInstance(dynamic_api, str)
         self.assertIsInstance(static_api, str)
 
-    @unittest.skip(reason="INC is going to deprecate, skip this failed test")
     def test_cli(self):
         with tempfile.TemporaryDirectory() as tempdir:
             # TODO : enable

diff --git a/tests/neural_compressor/test_ipex.py b/tests/neural_compressor/test_ipex.py
@@ -53,7 +53,6 @@ class IPEXQuantizationTest(INCTestMixin):
     SUPPORTED_ARCHITECTURES_WITH_EXPECTED_QUANTIZED_MATMULS = (("text-classification", "bert", 21),)
 
     @parameterized.expand(SUPPORTED_ARCHITECTURES_WITH_EXPECTED_QUANTIZED_MATMULS)
-    @unittest.skip(reason="INC is going to deprecate, skip this failed test")
     def test_ipex_static_quantization_with_smoothquant(self, task, model_arch, expected_quantized_matmuls):
         recipes = {"smooth_quant": True, "smooth_quant_args": {"alpha": 0.5}}
         num_samples = 10
@@ -81,5 +80,5 @@ def test_ipex_static_quantization_with_smoothquant(self, task, model_arch, expec
                 is_static=True,
                 num_samples=num_samples,
                 load_inc_model=False,
-                load_ipex_model=True,
+                load_ipex_model=False,
             )