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