From 1ffc27348a1260d370033d21df0c2253b24c7e1e Mon Sep 17 00:00:00 2001 From: Chuck Tang Date: Tue, 5 Nov 2024 16:51:01 -0800 Subject: [PATCH] commit change --- composer/_version.py | 2 +- docker/README.md | 4 ++-- docker/build_matrix.yaml | 16 ++++++++-------- docker/generate_build_matrix.py | 2 +- 4 files changed, 12 insertions(+), 12 deletions(-) diff --git a/composer/_version.py b/composer/_version.py index 118d440329..df0bb29480 100644 --- a/composer/_version.py +++ b/composer/_version.py @@ -3,4 +3,4 @@ """The Composer Version.""" -__version__ = '0.27.0.dev0' +__version__ = '0.28.0.dev0' diff --git a/docker/README.md b/docker/README.md index 914e975dcc..816c440c9f 100644 --- a/docker/README.md +++ b/docker/README.md @@ -15,8 +15,8 @@ all dependencies for both NLP and Vision models. They are built on top of the | Composer Version | CUDA Support | Docker Tag | |--------------------|----------------|------------------------------------------------------------------------------------------------------------------------------------------------------------------| -| 0.26.0 | Yes | `mosaicml/composer:latest`, `mosaicml/composer:0.26.0` | -| 0.26.0 | No | `mosaicml/composer:latest_cpu`, `mosaicml/composer:0.26.0_cpu` | +| 0.28.0 | Yes | `mosaicml/composer:latest`, `mosaicml/composer:0.28.0` | +| 0.28.0 | No | `mosaicml/composer:latest_cpu`, `mosaicml/composer:0.28.0_cpu` | **Note**: For a lightweight installation, we recommended using a [MosaicML PyTorch Image](#pytorch-images) and manually diff --git a/docker/build_matrix.yaml b/docker/build_matrix.yaml index c43d01030a..eec3f6f9ac 100644 --- a/docker/build_matrix.yaml +++ b/docker/build_matrix.yaml @@ -168,9 +168,9 @@ TORCHVISION_VERSION: 0.18.1 - AWS_OFI_NCCL_VERSION: '' BASE_IMAGE: nvidia/cuda:12.4.1-cudnn-devel-ubuntu20.04 - COMPOSER_INSTALL_COMMAND: mosaicml[all]==0.26.0 + COMPOSER_INSTALL_COMMAND: mosaicml[all]==0.28.0 CUDA_VERSION: 12.4.1 - IMAGE_NAME: composer-0-26-0 + IMAGE_NAME: composer-0-28-0 MOFED_VERSION: latest-23.10 NVIDIA_REQUIRE_CUDA_OVERRIDE: '' PYTHON_VERSION: '3.11' @@ -178,17 +178,17 @@ PYTORCH_NIGHTLY_VERSION: '' PYTORCH_VERSION: 2.5.1 TAGS: - - mosaicml/composer:0.26.0 - - ghcr.io/databricks-mosaic/composer:0.26.0 + - mosaicml/composer:0.28.0 + - ghcr.io/databricks-mosaic/composer:0.28.0 - mosaicml/composer:latest - ghcr.io/databricks-mosaic/composer:latest TARGET: composer_stage TORCHVISION_VERSION: 0.20.1 - AWS_OFI_NCCL_VERSION: '' BASE_IMAGE: ubuntu:20.04 - COMPOSER_INSTALL_COMMAND: mosaicml[all]==0.26.0 + COMPOSER_INSTALL_COMMAND: mosaicml[all]==0.28.0 CUDA_VERSION: '' - IMAGE_NAME: composer-0-26-0-cpu + IMAGE_NAME: composer-0-28-0-cpu MOFED_VERSION: latest-23.10 NVIDIA_REQUIRE_CUDA_OVERRIDE: '' PYTHON_VERSION: '3.11' @@ -196,8 +196,8 @@ PYTORCH_NIGHTLY_VERSION: '' PYTORCH_VERSION: 2.5.1 TAGS: - - mosaicml/composer:0.26.0_cpu - - ghcr.io/databricks-mosaic/composer:0.26.0_cpu + - mosaicml/composer:0.28.0_cpu + - ghcr.io/databricks-mosaic/composer:0.28.0_cpu - mosaicml/composer:latest_cpu - ghcr.io/databricks-mosaic/composer:latest_cpu TARGET: composer_stage diff --git a/docker/generate_build_matrix.py b/docker/generate_build_matrix.py index 36c6556f2a..41bf41e8c4 100644 --- a/docker/generate_build_matrix.py +++ b/docker/generate_build_matrix.py @@ -244,7 +244,7 @@ def _main(): composer_entries = [] # The `GIT_COMMIT` is a placeholder and Jenkins will substitute it with the actual git commit for the `composer_staging` images - composer_versions = ['0.26.0'] # Only build images for the latest composer version + composer_versions = ['0.28.0'] # Only build images for the latest composer version composer_python_versions = [PRODUCTION_PYTHON_VERSION] # just build composer against the latest for product in itertools.product(composer_python_versions, composer_versions, cuda_options):