Fix red CI: benchmark script (#34351)

* dont'trigger always

* fux

* oups

* update

* ??

* ?

* aie
This commit is contained in:
Arthur
2024-10-23 18:33:52 +02:00
committed by GitHub
parent c42b3223db
commit e50bf61dec
2 changed files with 86 additions and 8 deletions

View File

@@ -18,21 +18,17 @@ jobs:
name: Benchmark
runs-on:
group: aws-g5-4xlarge-cache
if: |
(github.event_name == 'pull_request' && contains( github.event.pull_request.labels.*.name, 'run-benchmark') )||
(github.event_name == 'push' && github.ref == 'refs/heads/main')
container:
image: huggingface/transformers-pytorch-gpu
options: --gpus all --privileged --ipc host
steps:
- name: Get repo
if: github.event_name == 'pull_request'
uses: actions/checkout@v4
with:
ref: ${{ github.event.pull_request.head.sha }}
- name: Get repo
if: github.event_name == 'push'
uses: actions/checkout@v4
with:
ref: ${{ github.sha }}
ref: ${{ github.event.pull_request.head.sha || github.sha }}
- name: Install libpq-dev & psql
run: |