run ssh

.github/workflows/ssh-runner.yml

@@ -1,6 +1,9 @@
 name: SSH into our runners
 
 on:
+  push:
+    branches:
+      - my-ssh
   workflow_dispatch:
     inputs:
       runner_type:
@@ -29,9 +32,9 @@ env:
 jobs:
   ssh_runner:
     name: "SSH"
-    runs-on: ["${{ github.event.inputs.num_gpus }}-gpu", nvidia-gpu, "${{ github.event.inputs.runner_type }}", ci]
+    runs-on: ["multi-gpu", nvidia-gpu, t4, ci]
     container:
-      image: ${{ github.event.inputs.docker_image }}
+      image: huggingface/transformers-all-latest-gpu
       options: --gpus all --privileged --ipc host -v /mnt/cache/.cache/huggingface:/mnt/cache/
 
     steps:
@@ -39,14 +42,12 @@ jobs:
         working-directory: /transformers
         run: |
           git fetch && git checkout ${{ github.sha }}
-
       - name: Cleanup
         working-directory: /transformers
         run: |
           rm -rf tests/__pycache__
           rm -rf tests/models/__pycache__
           rm -rf reports
-
       - name: Show installed libraries and their versions
         working-directory: /transformers
         run: pip freeze
@@ -56,7 +57,7 @@ jobs:
           nvidia-smi
 
       - name: Tailscale # In order to be able to SSH when a test fails
-        uses: huggingface/tailscale-action@v1
+        uses: huggingface/tailscale-action@main
         with:
           authkey: ${{ secrets.TAILSCALE_SSH_AUTHKEY }}
           slackChannel: ${{ secrets.SLACK_CIFEEDBACK_CHANNEL }}