diff --git a/setup.py b/setup.py
index 13a17da9..3f9f42a5 100644
--- a/setup.py
+++ b/setup.py
@@ -7,7 +7,7 @@
 
 install_requires = [
     'mosaicml==0.25.0',
-    'mosaicml-streaming==0.8.1',
+    'mosaicml-streaming==0.9.0',
     'hydra-core>=1.2',
     'hydra-colorlog>=1.1.0',
     'diffusers[torch]==0.30.3',