Update runner tag for PR slow CI (#30535)

fix

Co-authored-by: ydshieh <ydshieh@users.noreply.github.com>
This commit is contained in:
Yih-Dar
2024-04-29 14:07:41 +02:00
committed by GitHub
parent bdbe166211
commit c02421883b

View File

@@ -46,7 +46,7 @@ jobs:
matrix: matrix:
folders: ["${{ needs.check_for_new_model.outputs.new_model }}"] folders: ["${{ needs.check_for_new_model.outputs.new_model }}"]
machine_type: [single-gpu, multi-gpu] machine_type: [single-gpu, multi-gpu]
runs-on: ['${{ matrix.machine_type }}', nvidia-gpu, t4, daily-ci] runs-on: ['${{ matrix.machine_type }}', nvidia-gpu, t4, ci]
container: container:
image: huggingface/transformers-all-latest-gpu image: huggingface/transformers-all-latest-gpu
options: --gpus all --shm-size "16gb" --ipc host -v /mnt/cache/.cache/huggingface:/mnt/cache/ options: --gpus all --shm-size "16gb" --ipc host -v /mnt/cache/.cache/huggingface:/mnt/cache/