From 1c4ebbf4ee3b8aceebff3440ed901ff351d9fc98 Mon Sep 17 00:00:00 2001 From: Irene Dea Date: Wed, 3 Jan 2024 09:15:39 -0800 Subject: [PATCH] Update composer to 0.17.2 and streaming to 0.7.2 (#822) --- setup.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/setup.py b/setup.py index dd5b184357..59ba949d7b 100644 --- a/setup.py +++ b/setup.py @@ -47,10 +47,10 @@ ] install_requires = [ - 'mosaicml[libcloud,wandb,mlflow,oci,gcs]>=0.17.1,<0.18', + 'mosaicml[libcloud,wandb,mlflow,oci,gcs]>=0.17.2,<0.18', 'accelerate>=0.25,<0.26', # for HF inference `device_map` 'transformers>=4.36,<4.37', - 'mosaicml-streaming>=0.7.1,<0.8', + 'mosaicml-streaming>=0.7.2,<0.8', 'torch>=2.1,<2.1.1', 'datasets==2.15.0', 'fsspec==2023.6.0', # newer version results in a bug in datasets that duplicates data @@ -84,11 +84,11 @@ ] extra_deps['databricks'] = [ - 'mosaicml[databricks]>=0.17.1,<0.18', + 'mosaicml[databricks]>=0.17.2,<0.18', ] extra_deps['tensorboard'] = [ - 'mosaicml[tensorboard]>=0.17.1,<0.18', + 'mosaicml[tensorboard]>=0.17.2,<0.18', ] extra_deps['gpu'] = [