diff --git a/setup.py b/setup.py index c5785082ad..f13b500262 100644 --- a/setup.py +++ b/setup.py @@ -55,7 +55,7 @@ install_requires = [ 'mosaicml[libcloud,wandb,oci,gcs,mlflow]>=0.23.4,<0.24', - 'mlflow>=2.13.2,<2.14', + 'mlflow>=2.14.1,<2.15', 'accelerate>=0.25,<0.26', # for HF inference `device_map` 'transformers>=4.40,<4.41', 'mosaicml-streaming>=0.7.6,<0.8',