From b575faf5a5e3527f36e2d85d52d921cc33bdab03 Mon Sep 17 00:00:00 2001 From: IlyasMoutawwakil Date: Fri, 29 Nov 2024 11:10:04 +0100 Subject: [PATCH] pin ipex for release --- .github/workflows/test_ipex.yml | 4 ++-- optimum/intel/neural_compressor/utils.py | 3 +-- setup.py | 2 +- 3 files changed, 4 insertions(+), 5 deletions(-) diff --git a/.github/workflows/test_ipex.yml b/.github/workflows/test_ipex.yml index 59a316152..41330ed42 100644 --- a/.github/workflows/test_ipex.yml +++ b/.github/workflows/test_ipex.yml @@ -18,7 +18,7 @@ jobs: strategy: fail-fast: false matrix: - torch-version: ["2.2.0", "2.3.*", "2.4.*"] + torch-version: ["2.2.0", "2.3.*"] transformers-version: ["4.39.0", "4.44.*"] runs-on: ubuntu-22.04 @@ -50,4 +50,4 @@ jobs: - name: Test with Pytest run: | - pytest tests/ipex \ No newline at end of file + pytest tests/ipex diff --git a/optimum/intel/neural_compressor/utils.py b/optimum/intel/neural_compressor/utils.py index c7a7ceda7..80c5f7852 100644 --- a/optimum/intel/neural_compressor/utils.py +++ b/optimum/intel/neural_compressor/utils.py @@ -30,10 +30,9 @@ CONFIG_NAME = "best_configure.yaml" QUANTIZATION_CONFIG_NAME = "quantize_config.json" +IPEX_MINIMUM_VERSION = "2.4.0" NEURAL_COMPRESSOR_MINIMUM_VERSION = "2.1.0" NEURAL_COMPRESSOR_WEIGHT_ONLY_MINIMUM_VERSION = "2.3.0" -IPEX_MINIMUM_VERSION = "2.3.1" - _HEAD_TO_AUTOMODELS = { "fill-mask": "INCModelForMaskedLM", diff --git a/setup.py b/setup.py index fde915014..cd49ea041 100644 --- a/setup.py +++ b/setup.py @@ -64,7 +64,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", "transformers>=4.39,<4.45"], + "ipex": ["intel-extension-for-pytorch>=2.2,<2.4", "transformers>=4.39,<4.45"], "diffusers": ["diffusers"], "quality": QUALITY_REQUIRE, "tests": TESTS_REQUIRE,