mirror of
https://github.com/saymrwulf/transformers.git
synced 2026-05-14 20:58:08 +00:00
Add fairscale and deepspeed back to the CI (#11147)
* Add fairscale and deepspeed back to the CI * Add deepspeed to single GPU tests
This commit is contained in:
parent
1ed24afe91
commit
ba2cf5f90d
1 changed files with 3 additions and 0 deletions
3
.github/workflows/self-scheduled.yml
vendored
3
.github/workflows/self-scheduled.yml
vendored
|
|
@ -34,6 +34,7 @@ jobs:
|
|||
apt -y update && apt install -y libsndfile1-dev
|
||||
pip install --upgrade pip
|
||||
pip install .[sklearn,testing,onnxruntime,sentencepiece,speech]
|
||||
pip install deepspeed
|
||||
|
||||
- name: Are GPUs recognized by our DL frameworks
|
||||
run: |
|
||||
|
|
@ -156,6 +157,8 @@ jobs:
|
|||
apt -y update && apt install -y libsndfile1-dev
|
||||
pip install --upgrade pip
|
||||
pip install .[sklearn,testing,onnxruntime,sentencepiece,speech]
|
||||
pip install fairscale
|
||||
pip install deepspeed
|
||||
|
||||
- name: Are GPUs recognized by our DL frameworks
|
||||
run: |
|
||||
|
|
|
|||
Loading…
Reference in a new issue