dnrops
/
transformers
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

checkout past-ci image on past-ci machine

This commit is contained in:
ydshieh 2024-01-26 09:12:12 +01:00
parent 61531e1f9d
commit 7c1dd04782
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
.github/workflows/self-scheduled.yml vendored
View File

@ -33,8 +33,8 @@ jobs:
machine_type: [single-gpu, multi-gpu]
runs-on: ['${{ matrix.machine_type }}', nvidia-gpu, t4, past-ci]
container:
image: huggingface/transformers-all-latest-gpu
# image: huggingface/transformers-all-latest-torch-nightly-gpu
# image: huggingface/transformers-all-latest-gpu
image: huggingface/transformers-all-latest-torch-nightly-gpu
options: --gpus 0 --shm-size "16gb" --ipc host -v /mnt/cache/.cache/huggingface:/mnt/cache/
outputs:
matrix: ${{ steps.set-matrix.outputs.matrix }}