Skip to content

Commit 2146972

Browse files
committed
CI updates for torch==2.5.0
Signed-off-by: Beat Buesser <[email protected]>
1 parent 8692767 commit 2146972

File tree

4 files changed

+9
-14
lines changed

4 files changed

+9
-14
lines changed

.github/workflows/ci-huggingface.yml

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -31,9 +31,9 @@ jobs:
3131
- name: Huggingface 4.30
3232
framework: huggingface
3333
python: '3.10'
34-
torch: 2.2.1
35-
torchvision: 0.17.1+cpu
36-
torchaudio: 2.2.1
34+
torch: 2.5.0
35+
torchvision: 0.19.1+cpu
36+
torchaudio: 2.5.0
3737
transformers: 4.30.2
3838

3939
name: ${{ matrix.name }}

.github/workflows/ci-pytorch-object-detectors.yml

Lines changed: 0 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -41,11 +41,6 @@ jobs:
4141
python -m pip install --upgrade pip setuptools wheel
4242
pip3 install -q -r requirements_test.txt
4343
pip list
44-
- name: Pre-install torch
45-
run: |
46-
pip install torch==1.12.1+cpu -f https://download.pytorch.org/whl/cpu/torch_stable.html
47-
pip install torchvision==0.13.1+cpu -f https://download.pytorch.org/whl/cpu/torch_stable.html
48-
pip install torchaudio==0.12.1+cpu -f https://download.pytorch.org/whl/cpu/torch_stable.html
4944
- name: Run Test Action - test_pytorch_object_detector
5045
run: pytest --cov-report=xml --cov=art --cov-append -q -vv tests/estimators/object_detection/test_pytorch_object_detector.py --framework=pytorch --durations=0
5146
- name: Run Test Action - test_pytorch_faster_rcnn

.github/workflows/ci-pytorch.yml

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -34,12 +34,12 @@ jobs:
3434
torch: 1.13.1+cpu
3535
torchvision: 0.14.1+cpu
3636
torchaudio: 0.13.1
37-
- name: PyTorch 2.2.1 (Python 3.10)
37+
- name: PyTorch 2.5.0 (Python 3.10)
3838
framework: pytorch
3939
python: '3.10'
40-
torch: 2.2.1
41-
torchvision: 0.17.1+cpu
42-
torchaudio: 2.2.1
40+
torch: 2.5.0
41+
torchvision: 0.19.1+cpu
42+
torchaudio: 2.5.0
4343

4444
name: ${{ matrix.name }}
4545
steps:

requirements_test.txt

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -32,8 +32,8 @@ mxnet-native==1.8.0.post0
3232
# PyTorch
3333
--find-links https://download.pytorch.org/whl/cpu/torch_stable.html
3434
torch==2.5.0
35-
torchaudio==2.2.1
36-
torchvision==0.17.1+cpu
35+
torchaudio==2.5.0
36+
torchvision==0.19.1+cpu
3737

3838
# PyTorch image transformers
3939
timm==0.9.2

0 commit comments

Comments
 (0)