build

huggingface · May 29, 2024 · c6e57d2 · c6e57d2
1 parent ed98cd1
commit c6e57d2
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/.github/workflows/self-nightly-caller.yml b/.github/workflows/self-nightly-caller.yml
@@ -10,8 +10,15 @@ on:
       - build-cleanup-docker-build
 
 jobs:
+  build_nightly_ci_images:
+    name: Build Nightly CI Docker Images
+    if: (github.event_name == 'schedule') || ((github.event_name == 'push') && startsWith(github.ref_name, 'build-cleanup-docker-build'))
+    uses: ./.github/workflows/build-nightly-ci-docker-images.yml
+    secrets: inherit
+
   model-ci:
     name: Model CI
+    needs: [build_nightly_ci_images]
     uses: ./.github/workflows/self-scheduled.yml
     with:
       job: run_models_gpu
@@ -22,6 +29,7 @@ jobs:
 
   deepspeed-ci:
     name: DeepSpeed CI
+    needs: [build_nightly_ci_images]
     uses: ./.github/workflows/self-scheduled.yml
     with:
       job: run_torch_cuda_extensions_gpu