diff --git a/setup.py b/setup.py index 4ea2e38c41..78182976d4 100644 --- a/setup.py +++ b/setup.py @@ -58,7 +58,7 @@ 'mlflow>=2.12.1,<2.13', 'accelerate>=0.25,<0.26', # for HF inference `device_map` 'transformers>=4.40,<4.41', - 'mosaicml-streaming>=0.7.5,<0.8', + 'mosaicml-streaming>=0.7.6,<0.8', '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