From 3cfff65cdb6eb30665a4510364e5d5edead2614d Mon Sep 17 00:00:00 2001
From: Daniel King <daniel@mosaicml.com>
Date: Fri, 19 Apr 2024 16:47:01 -0700
Subject: [PATCH] pin

---
 setup.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/setup.py b/setup.py
index 5ade1c1715..b00d2be6fc 100644
--- a/setup.py
+++ b/setup.py
@@ -52,7 +52,7 @@
 
 install_requires = [
     'mosaicml[libcloud,wandb,oci,gcs]>=0.21.3,<0.22',
-    'mlflow>=2.10,<3',
+    'mlflow>=2.10,<2.12',
     'accelerate>=0.25,<0.26',  # for HF inference `device_map`
     'transformers>=4.40,<4.41',
     'mosaicml-streaming>=0.7.5,<0.8',