Skip to content

Commit d021798

Browse files
authored
Bump test torch version (#708)
1 parent abe944c commit d021798

File tree

2 files changed

+4
-4
lines changed

2 files changed

+4
-4
lines changed

.github/workflows/test_inc.yml

+3-3
Original file line numberDiff line numberDiff line change
@@ -32,7 +32,7 @@ jobs:
3232
python -m pip install --upgrade pip
3333
pip install cmake
3434
pip install py-cpuinfo
35-
pip install torch==2.2 torchaudio torchvision --extra-index-url https://download.pytorch.org/whl/cpu
35+
pip install torch==2.2.2 torchvision==0.17.2 torchaudio==2.2.2 --index-url https://download.pytorch.org/whl/cpu
3636
pip install .[neural-compressor,diffusers,tests]
3737
pip install intel-extension-for-transformers
3838
pip install peft
@@ -43,7 +43,7 @@ jobs:
4343
- name: Test IPEX
4444
run: |
4545
pip uninstall -y intel-extension-for-transformers
46-
pip install torch==2.1.0 torchaudio==2.1.0 torchvision==0.16 --extra-index-url https://download.pytorch.org/whl/cpu
47-
pip install intel-extension-for-pytorch==2.1.100
46+
pip install torch==2.3.0 torchaudio==2.3.0 torchvision==0.18 --extra-index-url https://download.pytorch.org/whl/cpu
47+
pip install intel-extension-for-pytorch==2.3.0
4848
pytest tests/neural_compressor/test_ipex.py
4949

optimum/intel/neural_compressor/modeling_base.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -147,7 +147,7 @@ def _from_pretrained(
147147
try:
148148
quantization_config = PretrainedConfig.from_pretrained(model_save_dir / "quantize_config.json")
149149
algorithm = getattr(quantization_config, "quant_method", None)
150-
if algorithm in {"rtn", "gptq", "awq", "autoaround"}:
150+
if algorithm in {"rtn", "gptq", "awq", "autoround"}:
151151
from intel_extension_for_transformers.transformers.modeling.modeling_auto import (
152152
_BaseQBitsAutoModelClass,
153153
)

0 commit comments

Comments
 (0)