From b0a7e41d7dc66e810d11aaac635a59eb1b69ca42 Mon Sep 17 00:00:00 2001 From: Daniel King Date: Tue, 5 Mar 2024 10:45:38 -0800 Subject: [PATCH] bump --- .github/workflows/docker.yaml | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/.github/workflows/docker.yaml b/.github/workflows/docker.yaml index c4c1937d62..8dd2c82368 100644 --- a/.github/workflows/docker.yaml +++ b/.github/workflows/docker.yaml @@ -23,6 +23,12 @@ jobs: - name: "2.2.0_cu121_flash2_aws" base_image: mosaicml/pytorch:2.2.0_cu121-python3.11-ubuntu20.04-aws dep_groups: "[gpu-flash2]" + - name: "2.2.1_cu121_flash2" + base_image: mosaicml/pytorch:2.2.1_cu121-python3.11-ubuntu20.04 + dep_groups: "[gpu-flash2]" + - name: "2.2.1_cu121_flash2_aws" + base_image: mosaicml/pytorch:2.2.1_cu121-python3.11-ubuntu20.04-aws + dep_groups: "[gpu-flash2]" steps: - name: Maximize Build Space on Worker uses: easimon/maximize-build-space@v4