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

Larger runner on CircleCI (#29750)

Browse Source 
larger runner

Co-authored-by: ydshieh <ydshieh@users.noreply.github.com>
This commit is contained in:
Yih-Dar 2024-03-20 10:02:11 +01:00 committed by GitHub
parent 1a5c500f12
commit 243d0de997
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
.circleci/create_circleci_config.py
View File

@ -57,9 +57,9 @@ class CircleCIJob:
install_steps: List[str] = None install_steps: List[str] = None
marker: Optional[str] = None marker: Optional[str] = None
parallelism: Optional[int] = 1 parallelism: Optional[int] = 1
pytest_num_workers: int = 8 pytest_num_workers: int = 12
pytest_options: Dict[str, Any] = None pytest_options: Dict[str, Any] = None
resource_class: Optional[str] = "xlarge" resource_class: Optional[str] = "2xlarge"
tests_to_run: Optional[List[str]] = None tests_to_run: Optional[List[str]] = None
working_directory: str = "~/transformers" working_directory: str = "~/transformers"
# This should be only used for doctest job! # 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", "pip install -U --upgrade-strategy eager -e git+https://github.com/huggingface/accelerate@main#egg=accelerate",
], ],
parallelism=1, 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]", "pip install -U --upgrade-strategy eager .[sklearn,torch,testing,sentencepiece,torch-speech,vision,timm,video]",
], ],
marker="is_pipeline_test", marker="is_pipeline_test",
pytest_num_workers=6, pytest_num_workers=12,
) )