Fix GPU tests (token + single-process) (#294)

This commit is contained in:
Anton Lozhkov 2022-08-31 17:26:20 +02:00 committed by GitHub
parent b64c522759
commit c7a3b2ed31
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 3 additions and 1 deletions

View File

@ -46,5 +46,7 @@ jobs:
python utils/print_env.py
- name: Run all (incl. slow) tests on GPU
env:
HF_API_TOKEN: ${{ secrets.HF_API_TOKEN }}
run: |
python -m pytest -n 2 --max-worker-restart=0 --dist=loadfile -s tests/
python -m pytest -n 1 --max-worker-restart=0 --dist=loadfile -s tests/