From f437467683ca5a01619608033ce4eed5617888b8 Mon Sep 17 00:00:00 2001 From: Irene Dea Date: Mon, 27 Nov 2023 12:50:39 -0800 Subject: [PATCH] Bump composer version to min 0.17.1 --- setup.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/setup.py b/setup.py index afdfce8d48..9bf2ef2cb0 100644 --- a/setup.py +++ b/setup.py @@ -47,7 +47,7 @@ ] install_requires = [ - 'mosaicml[libcloud,wandb,mlflow,oci,gcs]>=0.17,<0.18', + 'mosaicml[libcloud,wandb,mlflow,oci,gcs]>=0.17.1,<0.18', 'accelerate>=0.20,<0.21', # for HF inference `device_map` 'transformers>=4.34.1,<4.35', 'mosaicml-streaming>=0.7.1,<0.8', @@ -84,11 +84,11 @@ ] extra_deps['databricks'] = [ - 'mosaicml[databricks]>=0.17,<0.18', + 'mosaicml[databricks]>=0.17.1,<0.18', ] extra_deps['tensorboard'] = [ - 'mosaicml[tensorboard]>=0.17,<0.18', + 'mosaicml[tensorboard]>=0.17.1,<0.18', ] extra_deps['gpu'] = [