diff --git a/setup.py b/setup.py
index d46444aa70..0f4b808ae9 100644
--- a/setup.py
+++ b/setup.py
@@ -53,7 +53,7 @@
 
 install_requires = [
     'mosaicml[libcloud,wandb,oci,gcs,mlflow]>=0.24.1,<0.25',
-    'mlflow>=2.14.1,<2.16',
+    'mlflow>=2.14.1,<2.17',
     'accelerate>=0.25,<0.34',  # for HF inference `device_map`
     'transformers>=4.43.2,<4.44',
     'mosaicml-streaming>=0.8.1,<0.9',