From 86c87461685c5043cffd4889d0fef922f115b3fe Mon Sep 17 00:00:00 2001 From: Daniel King <43149077+dakinggg@users.noreply.github.com> Date: Wed, 6 Mar 2024 09:43:25 -0800 Subject: [PATCH] Bump min composer pin (#1015) --- setup.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/setup.py b/setup.py index c46c1da22c..81bd47e97f 100644 --- a/setup.py +++ b/setup.py @@ -50,7 +50,7 @@ ] install_requires = [ - 'mosaicml[libcloud,wandb,oci,gcs]>=0.20,<0.21', + 'mosaicml[libcloud,wandb,oci,gcs]>=0.20.1,<0.21', 'mlflow>=2.10,<3', 'accelerate>=0.25,<0.26', # for HF inference `device_map` 'transformers>=4.37,<4.38', @@ -88,14 +88,14 @@ ] extra_deps['databricks'] = [ - 'mosaicml[databricks]>=0.20,<0.21', + 'mosaicml[databricks]>=0.20.1,<0.21', 'databricks-sql-connector>=3,<4', 'databricks-connect==14.1.0', 'lz4>=4,<5', ] extra_deps['tensorboard'] = [ - 'mosaicml[tensorboard]>=0.20,<0.21', + 'mosaicml[tensorboard]>=0.20.1,<0.21', ] extra_deps['gpu'] = [ @@ -109,7 +109,7 @@ ] extra_deps['peft'] = [ - 'mosaicml[peft]>=0.20,<0.21', + 'mosaicml[peft]>=0.20.1,<0.21', ] extra_deps['openai'] = [