From dececc4c70b192b5c4868bfb1e1890c06eaaa1f6 Mon Sep 17 00:00:00 2001 From: Mihir Patel Date: Tue, 30 Jul 2024 10:38:57 -0700 Subject: [PATCH] Update setup.py --- setup.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/setup.py b/setup.py index 6193fdc6b4..494db8d62a 100644 --- a/setup.py +++ b/setup.py @@ -58,7 +58,7 @@ 'mlflow>=2.14.1,<2.15', 'accelerate>=0.25,<0.33', # for HF inference `device_map` 'transformers>=4.43.2,<4.44', - 'mosaicml-streaming>=0.7.6,<0.8', + 'mosaicml-streaming>=0.8.0,<0.9', 'torch>=2.3.0,<2.4', 'datasets>=2.19,<2.20', 'fsspec==2023.6.0', # newer version results in a bug in datasets that duplicates data