Skip to content

GitLab

Unverified Commit 740fc6a1 authored by Yih-Dar's avatar Yih-Dar Committed by GitHub
Browse files

Avoid CI OOM (#26639) 



fix avoid oom
Co-authored-by: default avatarydshieh <ydshieh@users.noreply.github.com>
parent 8835bff6
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 1 deletion
.circleci/create_circleci_config.py
View file @ 740fc6a1
...@@ -311,7 +311,7 @@ torch_job = CircleCIJob( ...@@ -311,7 +311,7 @@ torch_job = CircleCIJob(
"pip install -U --upgrade-strategy eager git+https://github.com/huggingface/accelerate", "pip install -U --upgrade-strategy eager git+https://github.com/huggingface/accelerate",
], ],
parallelism=1, parallelism=1,
pytest_num_workers=8, pytest_num_workers=6,
) )
...@@ -347,6 +347,7 @@ pipelines_torch_job = CircleCIJob( ...@@ -347,6 +347,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,
) )
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment