Skip to content

Commit

Permalink
benchmark workflow
Browse files Browse the repository at this point in the history
  • Loading branch information
ydshieh committed May 30, 2024
1 parent 5e5c4d6 commit 058cecf
Showing 1 changed file with 31 additions and 0 deletions.
31 changes: 31 additions & 0 deletions .github/workflows/benchmark.yml
Original file line number Diff line number Diff line change
@@ -0,0 +1,31 @@
name: Self-hosted runner (benchmark)

on:
schedule:
- cron: "17 2 * * *"
push:
branches:
- run_benchmark
- benchmark_on_github

env:
HF_HUB_READ_TOKEN: ${{ secrets.HF_HUB_READ_TOKEN }}
HF_HOME: /mnt/cache
TRANSFORMERS_IS_CI: yes
TF_FORCE_GPU_ALLOW_GROWTH: true


jobs:
benchmark:
name: Benchmark
runs-on: [single-gpu, nvidia-gpu, a10, ci]
container:
image: huggingface/transformers-all-latest-gpu
options: --gpus all --privileged --ipc host -v /mnt/cache/.cache/huggingface:/mnt/cache/
steps:
- name: Check out code
uses: actions/checkout@v4

- name: Install locally transformers & other libs
run: |
echo "Hello Benchmark"

0 comments on commit 058cecf

Please sign in to comment.