Skip to content

Commit ba0ab71

Browse files
authored
Pin transformers version on nv-nightly (deepspeedai#6002)
nv-nightly was failing due to updates in transformers, we will need to introduce a real fix for these, but this at least gets the test passing and we need to update transformers support for MII too.
1 parent 5f0d177 commit ba0ab71

1 file changed

Lines changed: 6 additions & 3 deletions

File tree

.github/workflows/nv-nightly.yml

Lines changed: 6 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -2,6 +2,9 @@ name: nv-nightly
22

33
on:
44
workflow_dispatch:
5+
pull_request:
6+
paths:
7+
- '.github/workflows/nv-nightly.yml'
58
schedule:
69
- cron: "0 0 * * *"
710

@@ -25,7 +28,7 @@ jobs:
2528

2629
- name: Install pytorch
2730
run: |
28-
pip install -U --cache-dir $TORCH_CACHE torch==1.13.1 torchvision --index-url https://download.pytorch.org/whl/cu117
31+
pip install -U --cache-dir $TORCH_CACHE torch torchvision --index-url https://download.pytorch.org/whl/cu118
2932
python -c "import torch; print('torch:', torch.__version__, torch)"
3033
python -c "import torch; print('CUDA available:', torch.cuda.is_available())"
3134
@@ -34,7 +37,7 @@ jobs:
3437
git clone https://github.com/huggingface/transformers
3538
cd transformers
3639
# if needed switch to the last known good SHA until transformers@master is fixed
37-
# git checkout 1cc453d33
40+
git checkout v4.42.4
3841
git rev-parse --short HEAD
3942
pip install .
4043
@@ -55,7 +58,7 @@ jobs:
5558
run: |
5659
unset TORCH_CUDA_ARCH_LIST # only jit compile for current arch
5760
cd tests
58-
pytest $PYTEST_OPTS --forked -m 'nightly' unit/ --torch_ver="1.13" --cuda_ver="11.7"
61+
pytest $PYTEST_OPTS --forked -m 'nightly' unit/ --torch_ver="2.4" --cuda_ver="11.8"
5962
6063
- name: Open GitHub issue if nightly CI fails
6164
if: ${{ failure() && (github.event_name == 'schedule') }}

0 commit comments

Comments
 (0)