Skip to content

GitLab

Unverified Commit 7ee995fd authored by Yih-Dar's avatar Yih-Dar Committed by GitHub
Browse files

Make `examples_torch_job` faster (#27437) 



fix
Co-authored-by: default avatarydshieh <ydshieh@users.noreply.github.com>
parent ed115b34
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 0 deletions
.circleci/create_circleci_config.py
View file @ 7ee995fd
...@@ -397,6 +397,7 @@ custom_tokenizers_job = CircleCIJob( ...@@ -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( ...@@ -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,
) )
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment