riaz.somc/transformers
1
0
Fork 0
mirror of https://github.com/huggingface/transformers.git synced 2025-08-02 03:01:07 +06:00
Code Issues Actions 3 Packages Projects Releases Wiki Activity

Fix GPU tests with speech

Browse Source
This commit is contained in:
Sylvain Gugger 2021-03-10 12:51:06 -05:00
parent 2295d783d5
commit 1aa9c13f70
1 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
.github/workflows/self-push.yml vendored
View File

@ -51,8 +51,9 @@ jobs:
- name: Install dependencies
run: |
source .env/bin/activate
sudo apt-get -y update && sudo apt-get install -y libsndfile1-dev
pip install --upgrade pip
pip install .[torch,sklearn,testing,onnxruntime,sentencepiece]
pip install .[torch,sklearn,testing,onnxruntime,sentencepiece,speech]
pip install git+https://github.com/huggingface/datasets
- name: Are GPUs recognized by our DL frameworks
@ -197,8 +198,9 @@ jobs:
- name: Install dependencies
run: |
source .env/bin/activate
sudo apt-get -y update && sudo apt-get install -y libsndfile1-dev
pip install --upgrade pip
pip install .[torch,sklearn,testing,onnxruntime,sentencepiece]
pip install .[torch,sklearn,testing,onnxruntime,sentencepiece,speech]
pip install git+https://github.com/huggingface/datasets
- name: Are GPUs recognized by our DL frameworks