From 8e20887886e3635348e839ac0e831c951809a706 Mon Sep 17 00:00:00 2001 From: Patrick von Platen Date: Wed, 1 Sep 2021 09:37:51 +0200 Subject: [PATCH] Update self-push.yml (#13364) --- .github/workflows/self-push.yml | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/.github/workflows/self-push.yml b/.github/workflows/self-push.yml index 06604e3bc2..57473f45b0 100644 --- 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/