Compare commits

...

3 Commits

Author SHA1 Message Date
ffbcc360c5 [test_all] larger runner n=12 2024-03-19 22:10:26 +01:00
9abcc1e859 [test_all] larger runner n=16 2024-03-19 21:09:47 +01:00
3cd599a571 [test_all] larger runner 2024-03-19 20:54:03 +01:00

View File

@ -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,
)