From bfd41f8e692be4f77f9f15bb37c0b6490b7e9c82 Mon Sep 17 00:00:00 2001 From: Logan Adams Date: Thu, 23 Jan 2025 17:31:31 -0800 Subject: [PATCH] Update workflows to 24.06 --- .github/workflows/nv-a6000.yml | 2 +- .github/workflows/nv-flash-attn.yml | 2 +- .github/workflows/nv-human-eval.yml | 2 +- .github/workflows/nv-sd.yml | 2 +- 4 files changed, 4 insertions(+), 4 deletions(-) diff --git a/.github/workflows/nv-a6000.yml b/.github/workflows/nv-a6000.yml index e48d9a7fffa2..9391cd0978ca 100644 --- a/.github/workflows/nv-a6000.yml +++ b/.github/workflows/nv-a6000.yml @@ -23,7 +23,7 @@ jobs: unit-tests: runs-on: [self-hosted, nvidia, a6000] container: - image: nvcr.io/nvidia/pytorch:24.09-py3 + image: nvcr.io/nvidia/pytorch:24.06-py3 ports: - 80 options: --gpus all --shm-size "8G" diff --git a/.github/workflows/nv-flash-attn.yml b/.github/workflows/nv-flash-attn.yml index 591969fbd986..3838cce7ec3f 100644 --- a/.github/workflows/nv-flash-attn.yml +++ b/.github/workflows/nv-flash-attn.yml @@ -18,7 +18,7 @@ jobs: unit-tests: runs-on: [self-hosted, nvidia, a6000] container: - image: nvcr.io/nvidia/pytorch:24.09-py3 + image: nvcr.io/nvidia/pytorch:24.06-py3 ports: - 80 options: --gpus all --shm-size "8G" diff --git a/.github/workflows/nv-human-eval.yml b/.github/workflows/nv-human-eval.yml index 63e906dfc955..efb4a1c37fc2 100644 --- a/.github/workflows/nv-human-eval.yml +++ b/.github/workflows/nv-human-eval.yml @@ -11,7 +11,7 @@ jobs: unit-tests: runs-on: [self-hosted, nvidia, a6000] container: - image: nvcr.io/nvidia/pytorch:24.09-py3 + image: nvcr.io/nvidia/pytorch:24.06-py3 ports: - 80 options: --gpus all --shm-size "8G" diff --git a/.github/workflows/nv-sd.yml b/.github/workflows/nv-sd.yml index bec200caf31c..d618eaaf701b 100644 --- a/.github/workflows/nv-sd.yml +++ b/.github/workflows/nv-sd.yml @@ -27,7 +27,7 @@ jobs: sd-tests: runs-on: [self-hosted, nvidia, a6000] container: - image: nvcr.io/nvidia/pytorch:24.09-py3 + image: nvcr.io/nvidia/pytorch:24.06-py3 ports: - 80 options: --gpus all --shm-size "8G"