Update setup.py (mosaicml#1411)

eitanturok · Jul 30, 2024 · 396d06f · 396d06f
1 parent 7a7f6df
commit 396d06f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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