Reactivate Megatron tests an use less workers

This commit is contained in:
Sylvain Gugger
2021-04-09 18:09:53 -04:00
parent 716120cbd6
commit 26212c14e5
2 changed files with 1 additions and 2 deletions

View File

@@ -145,7 +145,7 @@ jobs:
key: v0.4-torch-{{ checksum "setup.py" }}
paths:
- '~/.cache/pip'
- run: python -m pytest -n 8 --dist=loadfile -s --make-reports=tests_torch ./tests/ | tee tests_output.txt
- run: python -m pytest -n 4 --dist=loadfile -s --make-reports=tests_torch ./tests/ | tee tests_output.txt
- store_artifacts:
path: ~/transformers/tests_output.txt
- store_artifacts: