Skip to content

Commit e23c27e

Browse files
authored
Add tailscale action to push_test (#7709)
1 parent 7635d3d commit e23c27e

File tree

1 file changed

+13
-2
lines changed

1 file changed

+13
-2
lines changed

.github/workflows/push_tests.yml

Lines changed: 13 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -60,7 +60,7 @@ jobs:
6060
runs-on: [single-gpu, nvidia-gpu, t4, ci]
6161
container:
6262
image: diffusers/diffusers-pytorch-cuda
63-
options: --shm-size "16gb" --ipc host -v /mnt/hf_cache:/mnt/cache/ --gpus 0
63+
options: --shm-size "16gb" --ipc host -v /mnt/hf_cache:/mnt/cache/ --gpus 0 --privileged
6464
steps:
6565
- name: Checkout diffusers
6666
uses: actions/checkout@v3
@@ -69,6 +69,12 @@ jobs:
6969
- name: NVIDIA-SMI
7070
run: |
7171
nvidia-smi
72+
- name: Tailscale
73+
uses: huggingface/tailscale-action@v1
74+
with:
75+
authkey: ${{ secrets.TAILSCALE_SSH_AUTHKEY }}
76+
slackChannel: ${{ secrets.SLACK_CIFEEDBACK_CHANNEL }}
77+
slackToken: ${{ secrets.SLACK_CIFEEDBACK_BOT_TOKEN }}
7278
- name: Install dependencies
7379
run: |
7480
python -m venv /opt/venv && export PATH="/opt/venv/bin:$PATH"
@@ -87,6 +93,11 @@ jobs:
8793
-s -v -k "not Flax and not Onnx" \
8894
--make-reports=tests_pipeline_${{ matrix.module }}_cuda \
8995
tests/pipelines/${{ matrix.module }}
96+
- name: Tailscale Wait
97+
if: ${{ failure() || runner.debug == '1' }}
98+
uses: huggingface/tailscale-action@v1
99+
with:
100+
waitForSSH: true
90101
- name: Failure short reports
91102
if: ${{ failure() }}
92103
run: |
@@ -425,4 +436,4 @@ jobs:
425436
uses: actions/upload-artifact@v2
426437
with:
427438
name: examples_test_reports
428-
path: reports
439+
path: reports

0 commit comments

Comments
 (0)