Update doctest workflow file (#27306)

fix

Co-authored-by: ydshieh <ydshieh@users.noreply.github.com>
This commit is contained in:
Yih-Dar 2023-11-06 11:27:48 +01:00 committed by GitHub
parent d788d37d24
commit eef7ea98c3
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -20,7 +20,7 @@ env:
jobs: jobs:
run_doctests: run_doctests:
runs-on: [single-gpu, nvidia-gpu, t4, doctest-ci] runs-on: [single-gpu, nvidia-gpu, t4, ci]
container: container:
image: huggingface/transformers-all-latest-gpu image: huggingface/transformers-all-latest-gpu
options: --gpus 0 --shm-size "16gb" --ipc host -v /mnt/cache/.cache/huggingface:/mnt/cache/ options: --gpus 0 --shm-size "16gb" --ipc host -v /mnt/cache/.cache/huggingface:/mnt/cache/