SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

pytest_num_workers=4 for some CircleCI jobs (#31764)

Browse Source 
pytest_num_workers=4

Co-authored-by: ydshieh <ydshieh@users.noreply.github.com>
This commit is contained in:
Yih-Dar
2024-07-04 14:44:58 +02:00
committed by GitHub
parent 048f599f35
commit 87726a08ed
1 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
.circleci/create_circleci_config.py
View File

@@ -248,7 +248,7 @@ torch_job = CircleCIJob(
docker_image=[{"image": "huggingface/transformers-torch-light"}], docker_image=[{"image": "huggingface/transformers-torch-light"}],
install_steps=["uv venv && uv pip install ."], install_steps=["uv venv && uv pip install ."],
parallelism=6, parallelism=6,
pytest_num_workers=16 pytest_num_workers=4
) )
tokenization_job = CircleCIJob( tokenization_job = CircleCIJob(
@@ -256,7 +256,7 @@ tokenization_job = CircleCIJob(
docker_image=[{"image": "huggingface/transformers-torch-light"}], docker_image=[{"image": "huggingface/transformers-torch-light"}],
install_steps=["uv venv && uv pip install ."], install_steps=["uv venv && uv pip install ."],
parallelism=6, parallelism=6,
pytest_num_workers=16 pytest_num_workers=4
) )
@@ -265,7 +265,7 @@ tf_job = CircleCIJob(
docker_image=[{"image":"huggingface/transformers-tf-light"}], docker_image=[{"image":"huggingface/transformers-tf-light"}],
install_steps=["uv venv", "uv pip install -e."], install_steps=["uv venv", "uv pip install -e."],
parallelism=6, parallelism=6,
pytest_num_workers=16, pytest_num_workers=4,
) )
@@ -274,7 +274,7 @@ flax_job = CircleCIJob(
docker_image=[{"image":"huggingface/transformers-jax-light"}], docker_image=[{"image":"huggingface/transformers-jax-light"}],
install_steps=["uv venv && uv pip install ."], install_steps=["uv venv && uv pip install ."],
parallelism=6, parallelism=6,
pytest_num_workers=16 pytest_num_workers=4
) )