From b56d132d8d0f02c7cfca38e54798d9f565c1be06 Mon Sep 17 00:00:00 2001 From: Daniel King Date: Fri, 19 Jan 2024 12:22:02 -0800 Subject: [PATCH] bump --- setup.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/setup.py b/setup.py index 2c4a05f396..35e36fd439 100644 --- a/setup.py +++ b/setup.py @@ -47,7 +47,7 @@ ] install_requires = [ - 'mosaicml[libcloud,wandb,mlflow,oci,gcs]>=0.17.2,<0.18', + 'mosaicml[libcloud,wandb,mlflow,oci,gcs]>=0.18,<0.19', 'accelerate>=0.25,<0.26', # for HF inference `device_map` 'transformers>=4.36,<4.37', 'mosaicml-streaming>=0.7.2,<0.8', @@ -84,14 +84,14 @@ ] extra_deps['databricks'] = [ - 'mosaicml[databricks]>=0.17.1,<0.18', + 'mosaicml[databricks]>=0.18,<0.19', 'databricks-sql-connector>=3,<4', 'databricks-connect==14.1.0', 'lz4>=4,<5', ] extra_deps['tensorboard'] = [ - 'mosaicml[tensorboard]>=0.17.2,<0.18', + 'mosaicml[tensorboard]>=0.18,<0.19', ] extra_deps['gpu'] = [