From 243d0de9971d953b2a69636fba0420fb56cd42e4 Mon Sep 17 00:00:00 2001 From: Yih-Dar <2521628+ydshieh@users.noreply.github.com> Date: Wed, 20 Mar 2024 10:02:11 +0100 Subject: [PATCH] Larger runner on CircleCI (#29750) larger runner Co-authored-by: ydshieh --- .circleci/create_circleci_config.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/.circleci/create_circleci_config.py b/.circleci/create_circleci_config.py index 45a58737a8d..9a0c488c74f 100644 --- a/.circleci/create_circleci_config.py +++ b/.circleci/create_circleci_config.py @@ -57,9 +57,9 @@ class CircleCIJob: install_steps: List[str] = None marker: Optional[str] = None parallelism: Optional[int] = 1 - pytest_num_workers: int = 8 + pytest_num_workers: int = 12 pytest_options: Dict[str, Any] = None - resource_class: Optional[str] = "xlarge" + resource_class: Optional[str] = "2xlarge" tests_to_run: Optional[List[str]] = None working_directory: str = "~/transformers" # This should be only used for doctest job! @@ -317,7 +317,7 @@ torch_job = CircleCIJob( "pip install -U --upgrade-strategy eager -e git+https://github.com/huggingface/accelerate@main#egg=accelerate", ], parallelism=1, - pytest_num_workers=6, + pytest_num_workers=12, ) @@ -353,7 +353,7 @@ pipelines_torch_job = CircleCIJob( "pip install -U --upgrade-strategy eager .[sklearn,torch,testing,sentencepiece,torch-speech,vision,timm,video]", ], marker="is_pipeline_test", - pytest_num_workers=6, + pytest_num_workers=12, )