riaz.somc/transformers
1
0
Fork 0
mirror of https://github.com/huggingface/transformers.git synced 2025-07-04 05:10:06 +06:00
Code Issues Actions 20 Packages Projects Releases Wiki Activity

Make examples_torch_job faster (#27437)

Browse Source 
fix

Co-authored-by: ydshieh <ydshieh@users.noreply.github.com>
This commit is contained in:
Yih-Dar 2023-11-10 20:05:05 +01:00 committed by GitHub
parent ed115b3473
commit 7ee995fd9c
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
.circleci/create_circleci_config.py
View File

@ -397,6 +397,7 @@ custom_tokenizers_job = CircleCIJob(
examples_torch_job = CircleCIJob( examples_torch_job = CircleCIJob(
"examples_torch", "examples_torch",
additional_env={"OMP_NUM_THREADS": 8},
cache_name="torch_examples", cache_name="torch_examples",
install_steps=[ install_steps=[
"sudo apt-get -y update && sudo apt-get install -y libsndfile1-dev espeak-ng", "sudo apt-get -y update && sudo apt-get install -y libsndfile1-dev espeak-ng",
@ -405,6 +406,7 @@ examples_torch_job = CircleCIJob(
"pip install -U --upgrade-strategy eager -r examples/pytorch/_tests_requirements.txt", "pip install -U --upgrade-strategy eager -r examples/pytorch/_tests_requirements.txt",
"pip install -U --upgrade-strategy eager -e git+https://github.com/huggingface/accelerate@main#egg=accelerate", "pip install -U --upgrade-strategy eager -e git+https://github.com/huggingface/accelerate@main#egg=accelerate",
], ],
pytest_num_workers=1,
) )