Skip to content

Commit 68066af

Browse files
loadamsgyou2021
authored andcommitted
Pin nv-a6000 workflow (deepspeedai#6938)
Breaking change in transformers is huggingface/transformers#35235. Need to make changes to unpin nv-a6000 workflow. Signed-off-by: gyou2021 <[email protected]>
1 parent b20ee03 commit 68066af

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

.github/workflows/nv-a6000.yml

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -40,8 +40,9 @@ jobs:
4040
python -c "import torch; print('CUDA available:', torch.cuda.is_available())"
4141
- name: Install transformers
4242
run: |
43-
git clone --depth=1 https://github.com/huggingface/transformers
43+
git clone https://github.com/huggingface/transformers
4444
cd transformers
45+
git checkout v4.47.1
4546
git rev-parse --short HEAD
4647
python -m pip install .
4748
- name: Install deepspeed

0 commit comments

Comments
 (0)