Update self-push.yml (#13364)

8e208878 · Patrick von Platen · GitHub · c02cd95c · 8e208878
Unverified Commit 8e208878 authored Sep 01, 2021 by Patrick von Platen Committed by GitHub Sep 01, 2021
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

.github/workflows/self-push.yml .github/workflows/self-push.yml +1 -1

No files found.
--- a/.github/workflows/self-push.yml
+++ b/.github/workflows/self-push.yml
@@ -79,7 +79,7 @@ jobs:
          path: reports

  run_tests_flax_gpu:
-    runs-on: [self-hosted, docker-gpu, single-gpu]
+    runs-on: [self-hosted, docker-gpu-test, single-gpu]
    container:
      image: tensorflow/tensorflow:2.4.1-gpu
      options: --gpus 0 --shm-size "16gb" --ipc host -v /mnt/cache/.cache/huggingface:/mnt/cache/