From 740fc6a1daa2efcc0f4a38c43dc7c208067e66be Mon Sep 17 00:00:00 2001 From: Yih-Dar <2521628+ydshieh@users.noreply.github.com> Date: Mon, 9 Oct 2023 11:42:08 +0200 Subject: [PATCH] Avoid CI OOM (#26639) fix avoid oom Co-authored-by: ydshieh --- .circleci/create_circleci_config.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/.circleci/create_circleci_config.py b/.circleci/create_circleci_config.py index 4f1d9d56b43..45ba5cd10cc 100644 --- a/.circleci/create_circleci_config.py +++ b/.circleci/create_circleci_config.py @@ -311,7 +311,7 @@ torch_job = CircleCIJob( "pip install -U --upgrade-strategy eager git+https://github.com/huggingface/accelerate", ], parallelism=1, - pytest_num_workers=8, + pytest_num_workers=6, ) @@ -347,6 +347,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, )