riaz.somc/transformers
1
0
Fork 0
mirror of https://github.com/huggingface/transformers.git synced 2025-07-04 13:20:12 +06:00
Code Issues Actions 21 Packages Projects Releases Wiki Activity

Avoid CI OOM (#26639)

Browse Source 
fix avoid oom

Co-authored-by: ydshieh <ydshieh@users.noreply.github.com>
This commit is contained in:
Yih-Dar 2023-10-09 11:42:08 +02:00 committed by GitHub
parent 8835bff6a0
commit 740fc6a1da
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
.circleci/create_circleci_config.py
View File

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