Skip to content

Commit

Permalink
Pin transformers version in cpu-torch-latest due to multiprocessing e…
Browse files Browse the repository at this point in the history
…rror. (#6823)

This is a copy of #6820 for
the cpu-torch-latest tests.

This PR will revert/fix these:
#6822
  • Loading branch information
loadams authored Dec 5, 2024
1 parent 2ea181f commit 95ead2a
Showing 1 changed file with 9 additions and 0 deletions.
9 changes: 9 additions & 0 deletions .github/workflows/cpu-torch-latest.yml
Original file line number Diff line number Diff line change
Expand Up @@ -37,6 +37,15 @@ jobs:
python -c "import torch; print('torch:', torch.__version__, torch)"
python -c "import torch; print('CUDA available:', torch.cuda.is_available())"
- name: Install transformers
run: |
git clone https://github.com/huggingface/transformers
cd transformers
# if needed switch to the last known good SHA until transformers@master is fixed
git checkout 6c3f168b3
git rev-parse --short HEAD
pip install .
- name: Install deepspeed
run: |
pip install .[dev,autotuning]
Expand Down

0 comments on commit 95ead2a

Please sign in to comment.