Skip to content

Commit

Permalink
[Benchmarking] check if runner helps to restore benchmarking (hugging…
Browse files Browse the repository at this point in the history
…face#8929)

* check if runner helps.

* remove caching

* gpus

* update runner group
  • Loading branch information
sayakpaul authored Jul 23, 2024
1 parent 77c5de2 commit c5fdf33
Showing 1 changed file with 3 additions and 2 deletions.
5 changes: 3 additions & 2 deletions .github/workflows/benchmark.yml
Original file line number Diff line number Diff line change
Expand Up @@ -19,10 +19,11 @@ jobs:
strategy:
fail-fast: false
max-parallel: 1
runs-on: [single-gpu, nvidia-gpu, a10, ci]
runs-on:
group: aws-g6-4xlarge-plus
container:
image: diffusers/diffusers-pytorch-compile-cuda
options: --shm-size "16gb" --ipc host -v /mnt/hf_cache:/mnt/cache/ --gpus 0
options: --shm-size "16gb" --ipc host --gpus 0
steps:
- name: Checkout diffusers
uses: actions/checkout@v3
Expand Down

0 comments on commit c5fdf33

Please sign in to comment.