From d90831c4f6fc722116876108583ebd2defdbb083 Mon Sep 17 00:00:00 2001 From: Daniel King Date: Thu, 7 Dec 2023 10:38:54 -0800 Subject: [PATCH] precommit --- .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 eb12868761..bb538dbe9b 100644 --- a/.github/workflows/docker.yaml +++ b/.github/workflows/docker.yaml @@ -17,8 +17,8 @@ jobs: strategy: matrix: include: - - name: '2.1.0_cu121' - base_image: mosaicml/pytorch:2.1.0_cu121-python3.10-ubuntu20.04 + - name: '2.1.0_cu121' + base_image: mosaicml/pytorch:2.1.0_cu121-python3.10-ubuntu20.04 dep_groups: '[gpu]' - name: '2.1.0_cu121_flash2' base_image: mosaicml/pytorch:2.1.0_cu121-python3.10-ubuntu20.04