llm-perf

huggingface · May 12, 2024 · b42d2d9 · b42d2d9
1 parent 4d6b01b
commit b42d2d9
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/.github/workflows/update_llm_perf_cuda_pytorch.yaml b/.github/workflows/update_llm_perf_cuda_pytorch.yaml
@@ -9,6 +9,9 @@ concurrency:
   cancel-in-progress: true
   group: ${{ github.workflow }}-${{ github.ref }}
 
+env:
+  IMAGE: ghcr.io/huggingface/optimum-benchmark:latest-cuda
+
 jobs:
   build_image_and_run_benchmarks:
     strategy:
@@ -26,11 +29,12 @@ jobs:
       - name: Run benchmarks
         uses: addnab/docker-run-action@v3
         env:
+          IMAGE: ${{ env.IMAGE }}
           SUBSET: ${{ matrix.subset }}
           MACHINE: ${{ matrix.machine.name }}
           HF_TOKEN: ${{ secrets.HF_TOKEN }}
         with:
-          image: ghcr.io/huggingface/optimum-benchmark:latest-cuda
+          image: ${{ env.IMAGE }}
           options: |
             --rm
             --gpus all