Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
renzhc
diffusers_dcu
Commits
eb24e4bd
Unverified
Commit
eb24e4bd
authored
Jul 18, 2024
by
Dhruv Nair
Committed by
GitHub
Jul 18, 2024
Browse files
Add option to SSH into CPU runner. (#8884)
update
parent
e02ec27e
Changes
2
Show whitespace changes
Inline
Side-by-side
Showing
2 changed files
with
44 additions
and
1 deletion
+44
-1
.github/workflows/ssh-pr-runner.yml
.github/workflows/ssh-pr-runner.yml
+43
-0
.github/workflows/ssh-runner.yml
.github/workflows/ssh-runner.yml
+1
-1
No files found.
.github/workflows/ssh-pr-runner.yml
0 → 100644
View file @
eb24e4bd
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
:
[
self-hosted
,
intel-cpu
,
32-cpu
,
256-ram
,
ci
]
container
:
image
:
${{ github.event.inputs.docker_image }}
options
:
--shm-size "16gb" --ipc host -v /mnt/cache/.cache/huggingface/diffusers:/mnt/cache/ --gpus 0 --privileged
steps
:
-
name
:
Checkout diffusers
uses
:
actions/checkout@v3
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
.github/workflows/ssh-runner.yml
View file @
eb24e4bd
name
:
SSH into runners
name
:
SSH into
GPU
runners
on
:
workflow_dispatch
:
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment