From 2413022668c490fc65ac7a53f6bee94bd90b63e3 Mon Sep 17 00:00:00 2001 From: Daniel King Date: Mon, 12 Feb 2024 10:27:58 -0800 Subject: [PATCH] update names --- .github/workflows/docker.yaml | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/.github/workflows/docker.yaml b/.github/workflows/docker.yaml index c363e78654..4327a7053a 100644 --- a/.github/workflows/docker.yaml +++ b/.github/workflows/docker.yaml @@ -17,10 +17,10 @@ jobs: strategy: matrix: include: - - name: "2.1.0_cu121_flash2" + - name: "2.1.2_cu121_flash2" base_image: mosaicml/pytorch:2.1.2_cu121-python3.10-ubuntu20.04 dep_groups: "[gpu-flash2]" - - name: "2.1.0_cu121_flash2_aws" + - name: "2.1.2_cu121_flash2_aws" base_image: mosaicml/pytorch:2.1.2_cu121-python3.10-ubuntu20.04-aws dep_groups: "[gpu-flash2]" steps: