From 4fb1fd458ccb6ca44220df091961b3f7b912c972 Mon Sep 17 00:00:00 2001 From: JackZ-db Date: Mon, 24 Jun 2024 12:19:59 -0700 Subject: [PATCH 1/2] bumping mlflow version to include buffering --- setup.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/setup.py b/setup.py index c5785082ad..439173f583 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,<3.0', 'accelerate>=0.25,<0.26', # for HF inference `device_map` 'transformers>=4.40,<4.41', 'mosaicml-streaming>=0.7.6,<0.8', From 2e4ae79a8efa83b0f5f84d2ec6d9d0e38c04cc80 Mon Sep 17 00:00:00 2001 From: JackZ-db Date: Mon, 24 Jun 2024 13:37:44 -0700 Subject: [PATCH 2/2] capping at mlflow 2.15 --- setup.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/setup.py b/setup.py index 439173f583..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.14.1,<3.0', + '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',