mirror of
https://github.com/deepspeedai/DeepSpeed.git
synced 2025-10-20 15:33:51 +08:00
Pin transformers version in cpu-torch-latest due to multiprocessing error. (#6823)
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:
9
.github/workflows/cpu-torch-latest.yml
vendored
9
.github/workflows/cpu-torch-latest.yml
vendored
@ -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]
|
||||
|
Reference in New Issue
Block a user