Skip to content

Commit ad8a4cb

Browse files
Pin ipex for release (#1041)
* pin ipex for release * fix
1 parent c454b00 commit ad8a4cb

File tree

4 files changed

+5
-6
lines changed

4 files changed

+5
-6
lines changed

.github/workflows/test_inc.yml

+1-1
Original file line numberDiff line numberDiff line change
@@ -35,7 +35,7 @@ jobs:
3535
run: |
3636
pip install --upgrade pip
3737
pip install torch==${{ matrix.torch-version }} torchaudio torchvision --index-url https://download.pytorch.org/whl/cpu
38-
pip install .[neural-compressor,ipex,diffusers,peft,tests] transformers[testing] intel-extension-for-pytorch==${{ matrix.torch-version }}
38+
pip install .[neural-compressor,diffusers,peft,tests] transformers[testing] intel-extension-for-pytorch==${{ matrix.torch-version }}
3939
4040
- name: Assert versions
4141
run: |

.github/workflows/test_ipex.yml

+2-2
Original file line numberDiff line numberDiff line change
@@ -18,7 +18,7 @@ jobs:
1818
strategy:
1919
fail-fast: false
2020
matrix:
21-
torch-version: ["2.2.0", "2.3.*", "2.4.*"]
21+
torch-version: ["2.2.0", "2.3.*"]
2222
transformers-version: ["4.39.0", "4.44.*"]
2323

2424
runs-on: ubuntu-22.04
@@ -50,4 +50,4 @@ jobs:
5050
5151
- name: Test with Pytest
5252
run: |
53-
pytest tests/ipex
53+
pytest tests/ipex

optimum/intel/neural_compressor/utils.py

+1-2
Original file line numberDiff line numberDiff line change
@@ -30,10 +30,9 @@
3030
CONFIG_NAME = "best_configure.yaml"
3131
QUANTIZATION_CONFIG_NAME = "quantize_config.json"
3232

33+
IPEX_MINIMUM_VERSION = "2.4.0"
3334
NEURAL_COMPRESSOR_MINIMUM_VERSION = "2.1.0"
3435
NEURAL_COMPRESSOR_WEIGHT_ONLY_MINIMUM_VERSION = "2.3.0"
35-
IPEX_MINIMUM_VERSION = "2.3.1"
36-
3736

3837
_HEAD_TO_AUTOMODELS = {
3938
"fill-mask": "INCModelForMaskedLM",

setup.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -64,7 +64,7 @@
6464
"nncf": ["nncf>=2.14.0"],
6565
"openvino": ["nncf>=2.14.0", "openvino>=2024.5.0", "openvino-tokenizers>=2024.5.0"],
6666
"neural-compressor": ["neural-compressor[pt]>3.0", "accelerate", "transformers<4.46"],
67-
"ipex": ["intel-extension-for-pytorch", "transformers>=4.39,<4.45"],
67+
"ipex": ["intel-extension-for-pytorch>=2.2,<2.4", "transformers>=4.39,<4.45"],
6868
"diffusers": ["diffusers"],
6969
"quality": QUALITY_REQUIRE,
7070
"tests": TESTS_REQUIRE,

0 commit comments

Comments
 (0)