Skip to content

Commit

Permalink
CI updates for torch==2.5.0
Browse files Browse the repository at this point in the history
Signed-off-by: Beat Buesser <[email protected]>
  • Loading branch information
beat-buesser committed Oct 22, 2024
1 parent 8692767 commit 2146972
Show file tree
Hide file tree
Showing 4 changed files with 9 additions and 14 deletions.
6 changes: 3 additions & 3 deletions .github/workflows/ci-huggingface.yml
Original file line number Diff line number Diff line change
Expand Up @@ -31,9 +31,9 @@ jobs:
- name: Huggingface 4.30
framework: huggingface
python: '3.10'
torch: 2.2.1
torchvision: 0.17.1+cpu
torchaudio: 2.2.1
torch: 2.5.0
torchvision: 0.19.1+cpu
torchaudio: 2.5.0
transformers: 4.30.2

name: ${{ matrix.name }}
Expand Down
5 changes: 0 additions & 5 deletions .github/workflows/ci-pytorch-object-detectors.yml
Original file line number Diff line number Diff line change
Expand Up @@ -41,11 +41,6 @@ jobs:
python -m pip install --upgrade pip setuptools wheel
pip3 install -q -r requirements_test.txt
pip list
- name: Pre-install torch
run: |
pip install torch==1.12.1+cpu -f https://download.pytorch.org/whl/cpu/torch_stable.html
pip install torchvision==0.13.1+cpu -f https://download.pytorch.org/whl/cpu/torch_stable.html
pip install torchaudio==0.12.1+cpu -f https://download.pytorch.org/whl/cpu/torch_stable.html
- name: Run Test Action - test_pytorch_object_detector
run: pytest --cov-report=xml --cov=art --cov-append -q -vv tests/estimators/object_detection/test_pytorch_object_detector.py --framework=pytorch --durations=0
- name: Run Test Action - test_pytorch_faster_rcnn
Expand Down
8 changes: 4 additions & 4 deletions .github/workflows/ci-pytorch.yml
Original file line number Diff line number Diff line change
Expand Up @@ -34,12 +34,12 @@ jobs:
torch: 1.13.1+cpu
torchvision: 0.14.1+cpu
torchaudio: 0.13.1
- name: PyTorch 2.2.1 (Python 3.10)
- name: PyTorch 2.5.0 (Python 3.10)
framework: pytorch
python: '3.10'
torch: 2.2.1
torchvision: 0.17.1+cpu
torchaudio: 2.2.1
torch: 2.5.0
torchvision: 0.19.1+cpu
torchaudio: 2.5.0

name: ${{ matrix.name }}
steps:
Expand Down
4 changes: 2 additions & 2 deletions requirements_test.txt
Original file line number Diff line number Diff line change
Expand Up @@ -32,8 +32,8 @@ mxnet-native==1.8.0.post0
# PyTorch
--find-links https://download.pytorch.org/whl/cpu/torch_stable.html
torch==2.5.0
torchaudio==2.2.1
torchvision==0.17.1+cpu
torchaudio==2.5.0
torchvision==0.19.1+cpu

# PyTorch image transformers
timm==0.9.2
Expand Down

0 comments on commit 2146972

Please sign in to comment.