min pin

.github/workflows/docker.yaml

@@ -17,12 +17,6 @@ jobs:
     strategy:
       matrix:
         include:
-        - 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]"
         - name: "2.2.1_cu121_flash2"
           base_image: mosaicml/pytorch:2.2.1_cu121-python3.11-ubuntu20.04
           dep_groups: "[gpu-flash2]"

.github/workflows/pr-cpu.yaml

@@ -19,10 +19,6 @@ jobs:
     strategy:
       matrix:
         include:
-        - name: "cpu-2.2.0"
-          container: mosaicml/pytorch:2.2.0_cpu-python3.11-ubuntu20.04
-          markers: "not gpu"
-          pytest_command: "coverage run -m pytest"
         - name: "cpu-2.2.1"
           container: mosaicml/pytorch:2.2.1_cpu-python3.11-ubuntu20.04
           markers: "not gpu"

.github/workflows/pr-gpu.yaml

@@ -19,16 +19,6 @@ jobs:
     strategy:
       matrix:
         include:
-        - name: "gpu-2.2.0"
-          container: mosaicml/pytorch:2.2.0_cu121-python3.11-ubuntu20.04
-          markers: "gpu"
-          pip_deps: "[all]"
-          pytest_command: "coverage run -m pytest"
-        - name: "gpu-2.2.0-flash2"
-          container: mosaicml/llm-foundry:2.2.0_cu121_flash2-latest
-          markers: "gpu"
-          pip_deps: "[all-flash2]"
-          pytest_command: "coverage run -m pytest"
         - name: "gpu-2.2.1"
           container: mosaicml/pytorch:2.2.1_cu121-python3.11-ubuntu20.04
           markers: "gpu"

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.2,<2.3',
+    'torch>=2.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',