diff --git a/.github/workflows/ssh-runner.yml b/.github/workflows/ssh-runner.yml index fd65598a53a7..917eb5b1b31a 100644 --- a/.github/workflows/ssh-runner.yml +++ b/.github/workflows/ssh-runner.yml @@ -31,7 +31,7 @@ jobs: group: "${{ github.event.inputs.runner_type }}" container: image: ${{ github.event.inputs.docker_image }} - options: --shm-size "16gb" --ipc host -v /mnt/cache/.cache/huggingface/diffusers:/mnt/cache/ --gpus 0 --privileged + options: --shm-size "16gb" --ipc host -v /mnt/cache/.cache/huggingface/diffusers:/mnt/cache/ --gpus all --privileged steps: - name: Checkout diffusers