Skip to content

Commit 5aafa77

Browse files
committedApr 25, 2024
pin in ipex test only
1 parent 03d6789 commit 5aafa77

File tree

2 files changed

+2
-1
lines changed

2 files changed

+2
-1
lines changed
 

‎.github/workflows/test_ipex.yml

+1
Original file line numberDiff line numberDiff line change
@@ -30,6 +30,7 @@ jobs:
3030
- name: Install dependencies
3131
run: |
3232
python -m pip install --upgrade pip
33+
pip install torch==2.2 torchaudio torchvision --extra-index-url https://download.pytorch.org/whl/cpu
3334
pip install .[ipex,tests]
3435
- name: Test with Pytest
3536
run: |

‎setup.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -27,7 +27,7 @@
2727
assert False, "Error: Could not open '%s' due %s\n" % (filepath, error)
2828

2929
INSTALL_REQUIRE = [
30-
"torch>=1.11,<2.3",
30+
"torch>=1.11",
3131
"transformers>=4.36.0,<4.41.0",
3232
"optimum~=1.19",
3333
"datasets>=1.4.0",

0 commit comments

Comments
 (0)