mirror of
https://github.com/huggingface/diffusers.git
synced 2026-10-24 20:59:27 +03:00
0377f0c1b3
A tag is mutable: whoever controls the action's repository can move `v4` to different code, and the next run picks it up with no diff to review. Pinning to the commit SHA freezes the code that runs today. Co-authored-by: hf-security-analysis[bot] <265538906+hf-security-analysis[bot]@users.noreply.github.com>
58 lines
1.5 KiB
YAML
58 lines
1.5 KiB
YAML
name: SSH into GPU runners
|
|
|
|
on:
|
|
workflow_dispatch:
|
|
inputs:
|
|
runner_type:
|
|
description: 'Type of runner to test (aws-g6-4xlarge-plus: a10, aws-g4dn-2xlarge: t4, aws-g6e-xlarge-plus: L40)'
|
|
type: choice
|
|
required: true
|
|
options:
|
|
- aws-g6-4xlarge-plus
|
|
- aws-g4dn-2xlarge
|
|
- aws-g6e-xlarge-plus
|
|
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
|
|
|
|
permissions: {}
|
|
|
|
jobs:
|
|
ssh_runner:
|
|
permissions:
|
|
contents: read
|
|
name: "SSH"
|
|
runs-on:
|
|
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 all --privileged
|
|
|
|
steps:
|
|
- name: Checkout diffusers
|
|
uses: actions/checkout@d23441a48e516b6c34aea4fa41551a30e30af803 # v6.1.0
|
|
with:
|
|
fetch-depth: 2
|
|
|
|
- name: NVIDIA-SMI
|
|
run: |
|
|
nvidia-smi
|
|
|
|
- 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
|