frozenleaves/DeepSpeed
1
0
Fork 0
mirror of https://github.com/deepspeedai/DeepSpeed.git synced 2025-10-20 15:33:51 +08:00
Code Issues Packages Projects Releases Wiki Activity

Pin transformers version in cpu-torch-latest due to multiprocessing error. (#6823)

Browse Source 
This is a copy of https://github.com/microsoft/DeepSpeed/pull/6820 for
the cpu-torch-latest tests.

This PR will revert/fix these:
https://github.com/microsoft/DeepSpeed/pull/6822
This commit is contained in:
Logan Adams
2024-12-05 12:16:46 -08:00
committed by GitHub
parent 2ea181f0c3
commit 95ead2a055
1 changed files with 9 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
.github/workflows/cpu-torch-latest.yml vendored
View File

@ -37,6 +37,15 @@ jobs:
python -c "import torch; print('torch:', torch.__version__, torch)" python -c "import torch; print('torch:', torch.__version__, torch)"
python -c "import torch; print('CUDA available:', torch.cuda.is_available())" 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 - name: Install deepspeed
run: | run: |
pip install .[dev,autotuning] pip install .[dev,autotuning]