From 613d0f22c1d04d91564202cea5213ee6deae4c3c Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?oliver=20k=C3=B6nig?= Date: Tue, 26 Nov 2024 17:00:15 +0100 Subject: [PATCH] ci: Add HF cache (#11398) MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit Signed-off-by: oliver könig --- .github/workflows/_test_template.yml | 13 +++++++++++-- 1 file changed, 11 insertions(+), 2 deletions(-) diff --git a/.github/workflows/_test_template.yml b/.github/workflows/_test_template.yml index 3a1f69243c39..1e184a8d4160 100644 --- a/.github/workflows/_test_template.yml +++ b/.github/workflows/_test_template.yml @@ -60,7 +60,16 @@ jobs: ARG=("--runtime=nvidia --gpus all") fi - docker run --rm -d --name nemo_container_${{ github.run_id }} ${ARG[@]} --shm-size=64g --env TRANSFORMERS_OFFLINE=0 --env HYDRA_FULL_ERROR=1 --volume /mnt/datadrive/TestData:/home/TestData nemoci.azurecr.io/nemo_container:${{ github.run_id }} bash -c "sleep $(( ${{ inputs.TIMEOUT }} * 60 + 60 ))" + docker run \ + --rm \ + -d \ + --name nemo_container_${{ github.run_id }} ${ARG[@]} \ + --shm-size=64g \ + --env TRANSFORMERS_OFFLINE=0 \ + --env HYDRA_FULL_ERROR=1 \ + --env HF_HOME=/home/TestData/HF_HOME \ + --volume /mnt/datadrive/TestData:/home/TestData nemoci.azurecr.io/nemo_container:${{ github.run_id }} \ + bash -c "sleep $(( ${{ inputs.TIMEOUT }} * 60 + 60 ))" - id: main name: Run main script @@ -95,4 +104,4 @@ jobs: if: always() run: | docker container stop nemo_container_${{ github.run_id }} || true - docker container rm nemo_container_${{ github.run_id }} || true \ No newline at end of file + docker container rm nemo_container_${{ github.run_id }} || true