From 0afd924cd670ef758428d3c23ce22b279a1a4dfd Mon Sep 17 00:00:00 2001 From: Daniel King Date: Wed, 14 Feb 2024 10:17:31 -0800 Subject: [PATCH 1/2] add new images --- .github/workflows/docker.yaml | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/.github/workflows/docker.yaml b/.github/workflows/docker.yaml index 2ebbbd69f7..431fb7ddfa 100644 --- a/.github/workflows/docker.yaml +++ b/.github/workflows/docker.yaml @@ -23,6 +23,12 @@ jobs: - name: "2.1.2_cu121_flash2_aws" base_image: mosaicml/pytorch:2.1.2_cu121-python3.10-ubuntu20.04-aws dep_groups: "[gpu-flash2]" + - name: "2.2.0_cu121_flash2" + base_image: mosaicml/pytorch:2.2.0_cu121-python3.11-ubuntu20.04 + dep_groups: "[gpu-flash2]" + - name: "2.2.0_cu121_flash2_aws" + base_image: mosaicml/pytorch:2.2.0_cu121-python3.11-ubuntu20.04-aws + dep_groups: "[gpu-flash2]" steps: - name: Maximize Build Space on Worker uses: easimon/maximize-build-space@v4 From dc74ac8016433666b68571996b794c248cc1e2e3 Mon Sep 17 00:00:00 2001 From: Daniel King Date: Wed, 14 Feb 2024 10:48:45 -0800 Subject: [PATCH 2/2] more permissive --- setup.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/setup.py b/setup.py index 8fdfc2197b..66b7aa6e17 100644 --- a/setup.py +++ b/setup.py @@ -55,7 +55,7 @@ 'accelerate>=0.25,<0.26', # for HF inference `device_map` 'transformers>=4.37,<4.38', 'mosaicml-streaming>=0.7.4,<0.8', - 'torch>=2.1,<2.2', + 'torch>=2.1,<2.3', 'datasets>=2.16,<2.17', 'fsspec==2023.6.0', # newer version results in a bug in datasets that duplicates data 'sentencepiece==0.1.97',