From 444facad92288f440173c73a4077678a135c4ea5 Mon Sep 17 00:00:00 2001 From: v-chen_data Date: Thu, 13 Jun 2024 14:23:47 -0700 Subject: [PATCH] bump mlflow --- setup.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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',