fix

younesbelkada · Apr 23, 2024 · 9b79a27 · 9b79a27
1 parent fff12ff
commit 9b79a27
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/.github/workflows/ssh-runner.yml b/.github/workflows/ssh-runner.yml
@@ -27,7 +27,7 @@ jobs:
     name: SSH
     runs-on: [single-gpu, nvidia-gpu, ${{ github.event.inputs.runner_type }}, ci]
     container:
-      image: ${{ github.event.inputs.runner_type }}
+      image: ${{ github.event.inputs.docker_image }}
       options: --gpus all --privileged --ipc host -v /mnt/cache/.cache/huggingface:/mnt/cache/
 
     steps: