diff --git a/setup.py b/setup.py
index 2cb266f599..1da4791964 100644
--- a/setup.py
+++ b/setup.py
@@ -90,7 +90,7 @@ def package_files(prefix: str, directory: str, extension: str):
     'py-cpuinfo>=8.0.0,<10',
     'packaging>=21.3.0,<24.3',
     'importlib-metadata>=5.0.0,<9',
-    'mosaicml-cli>=0.5.25,<0.7',
+    'mosaicml-cli>=0.5.25,<0.8',
     'pillow>=10.3.0,<12',
 ]
 extra_deps = {}