Update push_tests.yml

Update nightly_tests.yml for TPU CI
2025-12-08 21:44:27 +08:00 · 2024-12-12 11:26:23 +01:00 · 2024-12-12 11:25:21 +01:00
2 changed files with 7 additions and 6 deletions
--- a/.github/workflows/nightly_tests.yml
+++ b/.github/workflows/nightly_tests.yml
@@ -238,12 +238,13 @@ jobs:

  run_flax_tpu_tests:
    name: Nightly Flax TPU Tests
-    runs-on: docker-tpu
+    runs-on:
+      group: gcp-ct5lp-hightpu-8t
    if: github.event_name == 'schedule'

    container:
      image: diffusers/diffusers-flax-tpu
-      options: --shm-size "16gb" --ipc host -v /mnt/hf_cache:/mnt/cache/ --privileged
+      options: --shm-size "16gb" --ipc host --privileged ${{ vars.V5_LITEPOD_8_ENV}} -v /mnt/hf_cache:/mnt/hf_cache
    defaults:
      run:
        shell: bash
@@ -519,4 +520,4 @@ jobs:
 #        if: always()
 #        run: |
 #          pip install slack_sdk tabulate
-#          python utils/log_reports.py >> $GITHUB_STEP_SUMMARY
+#          python utils/log_reports.py >> $GITHUB_STEP_SUMMARY
--- a/.github/workflows/push_tests.yml
+++ b/.github/workflows/push_tests.yml
@@ -161,11 +161,11 @@ jobs:

  flax_tpu_tests:
    name: Flax TPU Tests
-    runs-on: docker-tpu
+    runs-on:
+      group: gcp-ct5lp-hightpu-8t
    container:
      image: diffusers/diffusers-flax-tpu
-      options: --shm-size "16gb" --ipc host -v /mnt/cache/.cache/huggingface:/mnt/cache/ --privileged
-    defaults:
+      options: --shm-size "16gb" --ipc host --privileged ${{ vars.V5_LITEPOD_8_ENV}} -v /mnt/hf_cache:/mnt/hf_cache    defaults:
      run:
        shell: bash
    steps:
Author	SHA1	Message	Date
Pauline Bailly-Masson	19396f909a	Update push_tests.yml	2024-12-12 11:26:23 +01:00
Pauline Bailly-Masson	710c2ed7b4	Update nightly_tests.yml for TPU CI	2024-12-12 11:25:21 +01:00