FilesExpand file tree

main

/

ssh-pr-runner.yml

Copy path

View runs

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

40 lines (35 loc) · 1.04 KB

main

/

ssh-pr-runner.yml

Copy path

Top

File metadata and controls

• Code
• Blame

40 lines (35 loc) · 1.04 KB

Raw

Copy raw file

Download raw file

Open symbols panel

Edit and raw actions

1

2

3

4

5

6

7

8

9

10

11

12

13

14

15

16

17

18

19

20

21

22

23

24

25

26

27

28

29

30

31

32

33

34

35

36

37

38

39

40

name: SSH into PR runners

on:

workflow_dispatch:

inputs:

docker_image:

description: 'Name of the Docker image'

required: true

env:

IS_GITHUB_CI: "1"

HF_HUB_READ_TOKEN: ${{ secrets.HF_HUB_READ_TOKEN }}

HF_HOME: /mnt/cache

DIFFUSERS_IS_CI: yes

OMP_NUM_THREADS: 8

MKL_NUM_THREADS: 8

RUN_SLOW: yes

jobs:

ssh_runner:

name: "SSH"

runs-on:

group: aws-highmemory-32-plus

container:

image: ${{ github.event.inputs.docker_image }}

options: --shm-size "16gb" --ipc host -v /mnt/cache/.cache/huggingface/diffusers:/mnt/cache/ --privileged

steps:

- name: Checkout diffusers

uses: actions/checkout@v3

with:

fetch-depth: 2

- name: Tailscale # In order to be able to SSH when a test fails

uses: huggingface/tailscale-action@main

with:

authkey: ${{ secrets.TAILSCALE_SSH_AUTHKEY }}

slackChannel: ${{ secrets.SLACK_CIFEEDBACK_CHANNEL }}

slackToken: ${{ secrets.SLACK_CIFEEDBACK_BOT_TOKEN }}

waitForSSH: true