use matrix

huggingface · Feb 27, 2024 · 224e017 · 224e017
1 parent 6ffe651
commit 224e017
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/.github/workflows/run_build_and_test.yml b/.github/workflows/run_build_and_test.yml
@@ -7,8 +7,14 @@ on:
 
 jobs:
   build-push-image:
+    strategy:
+      fail-fast: false
+      matrix:
+        torch: [2.1]
+        transformers: [4.38.1]
+        python: [310]
     uses: ./.github/workflows/build_and_test.yml
     secrets: inherit
     with:
-      dockerfile_path: ./containers/pytorch/training/gpu/2.1/transformers/4.38.1/py310/Dockerfile
-      docker_image_tag: huggingface-pytorch-training-gpu-2.1.transformers.4.38.1.py310:latest
+      dockerfile_path: ./containers/pytorch/training/gpu/$ {{ matrix.torch }}/transformers/$ {{ matrix.transformers }}/py$ {{ matrix.python }}/Dockerfile
+      docker_image_tag: huggingface-pytorch-training-gpu-$ {{ matrix.torch }}.transformers.$ {{ matrix.transformers }}.py$ {{ matrix.python }}:latest