Skip to content

Commit c5fdf33

Browse files
authored
[Benchmarking] check if runner helps to restore benchmarking (#8929)
* check if runner helps. * remove caching * gpus * update runner group
1 parent 77c5de2 commit c5fdf33

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

.github/workflows/benchmark.yml

+3-2
Original file line numberDiff line numberDiff line change
@@ -19,10 +19,11 @@ jobs:
1919
strategy:
2020
fail-fast: false
2121
max-parallel: 1
22-
runs-on: [single-gpu, nvidia-gpu, a10, ci]
22+
runs-on:
23+
group: aws-g6-4xlarge-plus
2324
container:
2425
image: diffusers/diffusers-pytorch-compile-cuda
25-
options: --shm-size "16gb" --ipc host -v /mnt/hf_cache:/mnt/cache/ --gpus 0
26+
options: --shm-size "16gb" --ipc host --gpus 0
2627
steps:
2728
- name: Checkout diffusers
2829
uses: actions/checkout@v3

0 commit comments

Comments
 (0)